yolov5分割训练
准备数据集:
使用labelme对自己的数据进行分割得到json文件
使用转换代码转换成txt文件。
yolov5:
首先到yolov5的页面上git下代码
https://github.com/ultralytics/yolov5/blob/master/README.zh-CN.md
cd到想git到的目录开始git代码
git clone https://github.com/ultralytics/yolov5 # clone yolov5 cd yolov5 pip install -r requirements.txt # install
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple 使用清华源进行下载
本次训练路面分割,只有一个类别road 先去models/segment/yolov5s-seg.yaml
修改nc(类别数)为1

在coco128-seg中修改dataset根目录,训练集目录,测试集目录 class的名字

调参 主要调整weights,cfg,data,epoch,batchsize,img,device

根据模型调整hyp参数

到分割目录下
python train.py
训练结束后开始预测环节 调参

浙公网安备 33010602011771号