yolov5自己的数据集制作

文章参考博主：风吹落叶花飘荡

一、制作数据集

1、创建文件夹结构如下

2、将之前的图片以及标注数据放入mydata文件夹

test与train集合一般比例为2：8或3：7
训练的放在train,检测效果的放在test
下面labels的也是这样
注意：images和labels下的图片与标签要一一对应，所以图片标注前最好先重命名为1，2，3…
test的图片对应test的标签
train的图片对应train的标签

3、新建一个mydata.yaml文件

二、基于数据集训练模型

1、基于数据集训练模型

2、开始根据制作好的数据集训练模型

3、模型训练结束

注：由上图可以看到这次训练的模型保存在runs/train/expxxx

三、部署模型

在detect.py中调用模型

检测正常运行结果如下