错误指令:
python train.py --workers 2 --device 0 --batch-size 8 --data datasets/data.yaml --img 640 640 --cfg /cfg/training/yolov7.yaml --weights ‘’ --name yolov7 --hyp data/hyp.scratch.p5.yaml --epochs 150
其实是换了自己的数据集之后,yolov7的配置文件写错了。
按照这个作者的重写一遍就可以训练了。
【小白教学】如何用YOLOv7训练自己的数据集
配置好之后的指令:
python train.py --workers 2 --device 0 --batch-size 16 --weights weights/yolov7-tiny.pt --cfg cfg/training/yolov7-tiny.yaml --data data/data.yaml --epoch 150
文章讲述了作者在使用YOLOv7时遇到的错误,主要是配置文件问题。后文提供了正确配置后的训练指令,包括修改batch-size和加载预训练权重。
1918

被折叠的 条评论
为什么被折叠?



