1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > (四)目标检测yolov5-6.0模型训练

(四)目标检测yolov5-6.0模型训练

时间:2021-10-14 10:33:51

相关推荐

(四)目标检测yolov5-6.0模型训练

安装好yolov5-6.0环境后

1.数据结构格式

文件夹结构:新建coco128

images文件夹下的train与val都是图像

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

2.修改配置文件

第一个需要修改的文件data/coco128.yaml

修改数据集路径:

例如:datasets这个文件夹是我自己加的,可以不加

修改类别数、类别

第二个需要修改的文件: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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。