yolov5训练自己的数据进行水印检测
文章目录
前言
从整体流程把握yolov5使用自己的数据进行目标检测,防止同学们踩坑
一、下载yolov5源码
在yolov5官方git地址下载源码https://github.com/ultralytics/yolov5,确定好小版本,这样训练和部署模型就不会出现版本不一致的问题了
这里我选用的是v6.0版本,部署起来比较稳定
二、修改自己的yolov5的配置信息
1.修改data目录里面的配置文件
针对VOC.yaml文件进行修改
这是官方给的示例文件,告诉你输入模型的数据应该处理成什么样子
上图是我们修改好的配置文件
需要修改的是nc 改成你的目标检测类别数目
names 改成你自己的类目
train和val 是你自己的训练集和测试集的数据目录
其他的部分进行删除
输入模型的数据要处理成如下形式:
images是图片的数据
labels是标签信息数据
每个txt文件里面是检测目标的类别和坐标位置信息
2.修改models配置文件
看你具体使用哪个模型,使用哪个模型就修改哪个模型的配置文件
我这次采用的是yolov5s模型,所以修改yolov5s配置文件
修改的是nc数值,改成你目标检测的类别总数
修改完以后配置部分就完成了接下来进行train.py函数进行修改
3.修改train函数
需要修改的就是这个部分
分别是wieghts,cfg,data,epoch,batchsize
分别是预训练权重文件,模型配置信息,数据配置信息,训练轮次,批次数据
总结
以上就是使用自己的数据进行yolov5目标检测模型的训练流程了