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

训练结束后开始预测环节 调参

 

 

posted @ 2022-12-16 11:56  nerrr  阅读(711)  评论(0)    收藏  举报