Yolov2在服务器上训练自己的数据(一)

参考博客:https://timebutt.github.io/static/how-to-train-yolov2-to-detect-custom-objects/ 

                YOLO官网:https://pjreddie.com/darknet/yolo/

1.下载数据集到Windows,(此数据集是已经生成好的)

2.Getting Darknet:获得darknet的配置文件

3.生成test.txt以及train.txt(里面存储的是图片路径+名称)

4.创建自己的obj.data,obj.names以及yolo-obj.cfg

5.修改obj.data的数据(更改成自己的train以及test的路径),修改obj.names(检测的目标种类),修改配置文件(batch,subdivisions,classes,filters),下载我们所需要的pre-training的参数,以便于加快收敛速度

6.修改配置文件:GPU,OPENCV,CUDNN以及nvcc

7.进行训练数据:windows:darknet.exe  detector  train  cfg/obj.data  cfg/yolo-obj.cfg 预训练的参数文件

                           ubuntu:./darknet detector train  cfg/obj.data  cfg/yolo-obj.cfg 预训练的参数文件

8进行测试数据:ubuntu:./darknet detector test cfg/obj.data cfg/yolo-obj.cfg 选择训练好的参数文件

9.本地与服务器使用了scp命令:(windows)将本地文件上传到服务器需要先cd到该文件下,scp 文件 root@。。。:放置的绝对路径 (必须先cd到该文件,否则会出现错误).

                                                    将服务器文件下载到本地需要要针对当前的路径后面添加部分路径,scp -r root@...:源文件绝对路径  windows相对于当前路径的部分路径(比如当前路径是C:/users/xiaomi/Desktop 那么部分路径是就是文件的位置(比如是123文件夹)123/)

10.Ctrl+c取消输入路径

11.通常我们跑小程序可以在自己的终端中跑,如果是训练网络这样的大工作量事物,我们应该放在后台跑,这样既不耽误我们做以后的工作又可以防止我们因为电脑卡死等外部因素影响训练。

将程序放置在后台执行:nohup 命令 &即可,会在当前目录生成日志文件,我们可以通过访问日志了解训练信息。

关闭后台执行的程序时,jobs -l查看运行的后台进程,可以查看其进程号,使用kill -9 进程号即可关闭后台执行的程序。

 

### 选择适合YOLO模型训练服务器配置 对于运行YOLO系列模型特别是较新的版本如YOLOv8,硬件资源的选择至关重要。考虑到模型训练过程中对计算能力的需求较高,建议选用配备高性能GPU的云服务器实例来加速训练过程[^1]。 #### GPU性能要求 为了确保高效的模型训练体验,在选择GPU时应优先考虑NVIDIA Tesla V100、A100或类似的高端显卡型号。这些设备具备强大的浮点运算能力和较大的显存容量,能够显著缩短训练所需的时间并提高实验效率[^2]。 #### CPU与内存搭配 除了GPU外,CPU的核心数以及系统的RAM大小也会影响整体表现。推荐至少拥有16核以上的处理器配合不低于64GB RAM的工作站级设置,以便更好地支持多线程数据加载和其他辅助任务的同时执行[^3]。 #### 存储方案设计 鉴于深度学习项目通常涉及大量文件读写操作,因此快速可靠的磁盘访问速度同样不可忽视。可以选择SSD作为主要存储介质,并预留充足的空间用于保存原始素材、中间产物及最终成果等各类资料[^4]。 --- 针对上述需求特点,以下是几家主流的服务商及其提供的适配选项: | **服务商名称** | **推荐套餐示例** | | --- | --- | | 华为云 | Flexus云服务器X实例,提供多种规格可选,满足不同规模的数据处理需求;特别适用于AI应用场景下的弹性扩展特性。| | AWS (亚马逊网络服务) | EC2 P4d 实例类型,内置8颗 NVIDIA A100 Tensor Core GPUs 和高达1TB NVMe SSD本地存储空间,非常适合大规模分布式训练作业。| | Google Cloud Platform | N1 Preemptible VMs with TPU v3 Pods 或者最新的A2系列虚拟机,前者成本效益高而后者则提供了更强劲的算力输出。| ```bash # 示例命令:通过AWS CLI启动EC2实例 aws ec2 run-instances \ --image-id ami-xxxxxxxxxx \ --count 1 \ --instance-type p4d.24xlarge \ --key-name MyKeyPair \ --security-group-ids sg-xxxxxxxxxxxxxxx \ --subnet-id subnet-xxxxxxxxxxxxxxxxx ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值