手把手教你YOLOv5 训练自己的数据集

首先假设各位已经配好环境，下面开始一步一步操作

1.在yolov5/data/这个路径下有这样一些.yaml的配置文件：

2.打开coco128.yaml你会发现如下内容：

这个是由官方repo提供的配置脚本

download为自动下载数据集的地址

train为训练集目录

val为验证集目录

nc为数据集里类别的个数

names为类别名称

3.转换数据集格式

博主当前的数据集格式为VOC格式,文件架构如下

my-dataset/images：图片

my-dataset/xml：图片对应xml标注文件

把数据集转成yolov5所需要的格式，以coco128为例即

coco128/images/train:图片

coco128/labels/train:图片对应标注txt格式

第一列表示类别对应id，后面几列分别表示x_center y_center width height

根据如下脚本即可完成数据集转换：https://blog.youkuaiyun.com/qianchenghaozi/article/details/110523327

4.复制coco128.yaml修改为自己的数据集格式my-dataset.yaml运行

$ python train.py --img 640 --batch 16 --epochs 5 --data ./data/my-dataset.yaml --cfg ./models/yolov5s.yaml --weights ''