栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

目标检测yolov5-6.0模型训练

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

目标检测yolov5-6.0模型训练

安装好yolov5-6.0环境后 1.数据结构格式

文件夹结构:新建coco128

images文件夹下的train与val都是图像
labels文件夹下的train与val都是yolo格式的txt标签文件

2.修改配置文件

第一个需要修改的文件data/coco128.yaml
修改数据集路径:
例如:datasets这个文件夹是我i自己加的,可以不加

修改类别数、类别

第二个需要修改的文件:models/yolov5x.yaml
在yolov5文件目录下的models下,选择预训练权重文件的配置文件。我比较喜欢选择yolov5x.yaml。你也可选择别的。

只需要修改一下类别数即可

3.训练命令

根据你的显卡修改batch-size大小(一组几张图的数量)
训练轮数:epochs
多路读取数据:workers

python train.py --data data/coco128.yaml --cfg models/yolov5x.yaml --weights yolov5x.pt --batch-size 2 --epochs 20 --workers 2

训练结果在runs文件夹下面:

best.py 是训练最好的权重
last.pt 是最新训练的权重
我一般选择最好的best.py

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/969027.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号