文章目录
- 一、制作数据集
- 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中调用模型
检测正常运行结果如下