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
训练结束后开始预测环节 调参
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· AI 智能体引爆开源社区「GitHub 热点速览」
· 从HTTP原因短语缺失研究HTTP/2和HTTP/3的设计差异
· 三行代码完成国际化适配,妙~啊~