Yolov5训练制定数据集-完整版

1 创建数据集

1.1 创建如下文件夹

1.2 标注数据

标注数据集传送门
将结果按照images和Labels放到1.1中对应的文件中

1.3 创建custom_data.yaml文件

找到1 ，2 ，3步骤，找到data文件夹下的coco128.yaml，复制一份，改名为custom_data.yaml，将custom_data.yaml中添加3步骤的内容。

2 训练

python train.py --img 640 --batch 2 --epochs 100 --data custom_data.yaml --weights yolov5s.pt --nosave --cache

3 训练过程遇到的问题

3.1 可以会出现 “OSError: [WinError 1455] 页面文件太小”。

原因:python所在磁盘的虚拟内存不够。
解决方案：
（一）方法一：
将yolov5中utils文件下dataset.py的num_workers修改为0，如下图所示：

(2）给python所在盘分配虚拟内存
“查看高级系统配置”

3.2 opencv-python>=4.1.2 not found

pip install opencv-python 自动下载最新版本的

3.3 wandb.errors.CommError: check_hostname requires server_hostname

# 方框注册，然后会获得一个key
wandb init 

解决方案 ： 科学上网关掉！！！

3.3 Can’t get attribute ‘SPPF’ on <module ‘models.common’>

解决方案：文件权重问题，使用程序原有的权重文件。