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 @   nerrr  阅读(677)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~
点击右上角即可分享
微信分享提示