yolov5自己的数据集制作

自己的数据集制作"/>

yolov5自己的数据集制作

文章目录

• 一、制作数据集
• • 1、创建文件夹结构如下
  • 2、将之前的图片以及标注数据放入mydata文件夹
  • 3、新建一个mydata.yaml文件
• 二、基于数据集训练模型
• • 1、基于数据集训练模型
  • 2、开始根据制作好的数据集训练模型
  • 3、模型训练结束
• 三、部署模型

一、制作数据集

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中调用模型

检测正常运行结果如下