训练前准备
- 下载模型,地址:https://github.com/ultralytics/yolov5/tree/v7.0,下载7.0即可
- 解压缩后如图所示
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