自学习记录（1）--YOLOv7训练自己的数据集

1、下载源代码地址：https://github.com/WongKinYiu/yolov7 

2、首先需将VOC数据集（XML）转为YOLO格式，转换代码参考：VOC格式数据集转yolo格式数据集的方法_佐咖的博客-CSDN博客_voc转yolo格式

3、数据集格式同YOLOv5

4、训练

（1）data/自己名字.yaml

（2）cfg/training/yolov7.yaml中，nc需改

（3）train.py

(4)运行train.py

CPU：python train.py --workers 2 --device cpu --batch-size 8 --data data/mydata.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml 

 GPU：python train.py --workers 2 --device 0 --batch-size 8 --data data/mydata.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml 

5、遇到的问题

在自己笔记本可以跑通

1、使用Colab时遇到问题

（1）TypeError: dump_all() got an unexpected keyword argument 'sort_keys’

报错原因：这是因为pyyaml版本过低，更新一下就好了
解决方法：

 !pip install -U pyyaml

（2）assert img is not None, 'Image Not Found ' + path

报错原因：数据集路径一直找不到

解决方法：进入到其文件目录下去执行代码，cd前需是%，不能！

将labels/train2017.org和val2017.org删掉，重新生成