文章讲述了作者在使用YOLOv7时遇到的错误，主要是配置文件问题。后文提供了正确配置后的训练指令，包括修改batch-size和加载预训练权重。
错误指令：
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
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
