yolov5训练自己的水印数据进行水印目标检测

yolov5训练自己的数据进行水印检测

前言

从整体流程把握yolov5使用自己的数据进行目标检测，防止同学们踩坑

一、下载yolov5源码

在yolov5官方git地址下载源码https://github.com/ultralytics/yolov5，确定好小版本，这样训练和部署模型就不会出现版本不一致的问题了

这里我选用的是v6.0版本，部署起来比较稳定

二、修改自己的yolov5的配置信息

1.修改data目录里面的配置文件

针对VOC.yaml文件进行修改

这是官方给的示例文件，告诉你输入模型的数据应该处理成什么样子

上图是我们修改好的配置文件
需要修改的是nc 改成你的目标检测类别数目
names 改成你自己的类目
train和val 是你自己