【实例分割-yoloV5-7.0】使用自己的数据集训练yoloV5-7.0

训练前准备

1. 下载模型，地址：https://github.com/ultralytics/yolov5/tree/v7.0，下载7.0即可
2. 解压缩后如图所示

3.安装环境，根据requiremens.txt中环境安装。PS部分环境如gitpython不安装也可以正常运行
4.标注数据，使用labelme标注出来的数据为json格式，需要转换为txt格式
数据转换参考：https://blog.csdn.net/m0_64298393/article/details/134733964?spm=1001.2014.3001.5501
转后将数据标签及路径都知道data目录下的yaml文件，如下图所示

5.*打开segment下的train.py，千万不要直接用yolo目录下的train.py，因为yolo目录下的train.py只能识别矩形框，无法识别多边形
根据自己的路径依次配置：预训练权重（–weights），网络结构（–cfg），数据目录（–data，就是之前设置修改的yaml文件路径），其他如batchszie，imgsize根据自己需要设置

6.开始训练即可
7.开始训练后可在runs文件夹下找到训练文件

PS：预训练权重下载地址：https://github.com/ultralytics/yolov5/tree/v7.0