yolov5自己的数据集制作

文章目录

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

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

一、制作数据集

1、创建文件夹结构如下

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文件

yolov5自己的数据集制作_第2张图片

二、基于数据集训练模型

1、基于数据集训练模型

yolov5自己的数据集制作_第3张图片

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

yolov5自己的数据集制作_第4张图片

3、模型训练结束

yolov5自己的数据集制作_第5张图片
注:由上图可以看到这次训练的模型保存在runs/train/expxxx

三、部署模型

yolov5自己的数据集制作_第6张图片
在detect.py中调用模型
yolov5自己的数据集制作_第7张图片
检测正常运行结果如下
yolov5自己的数据集制作_第8张图片

你可能感兴趣的:(yolo,YOLO)