论文地址：https://arxiv.org/abs/2207.02696
源码地址：https://github.com/WongKinYiu/yolov7
一、数据集
直接把YOLOV5的数据集复制到根目录下
Images文件夹中是图片，Labels文件夹中是对应的标签
二、下载预训练权重
对应的模型下载对应的预训练权重，并放置根目录下，本文用的是yolov7.pt
三、配置yaml文件
在yolov7/data文件夹下新建VOC.yaml文件(直接复制COCO.yaml)
1、数据集地址改为对应的地址
2、更改数据集类别数和类别名称
四、修改train.py
epochs：训练的轮数
batch_size：YOLOV7占用显存较大，建议设置为2或4