Yolov7训练与测试

需要详细教程的可以直接参考这里：【小白教学】如何用YOLOv7训练自己的数据集

1.Yolov7训练

1.1模型下载

在GitHub上搜索Yolov7，可以直接下载公开的模型：Yolov7模型

1.2环境配置

Yolov7模型环境与Yolov5相同，不再赘述。如果需要可以参考这篇文章：【小白教学】如何用YOLOv7训练自己的数据集

1.3模型训练

1.3.1数据集制作

yolo数据集格式：Yolo格式的数据集介绍

1.3.2修改配置文件

修改./cfg/training/下所需训练模型的配置文件：

以yolov7.yaml为例，修改分类数：

修改./data/coco.yaml文件为如下格式：

train后面对应训练集图片位置，val后面对应验证集图片位置，nc后面对应分类数，name后面对应分类名称。

1.3.3模型训练

打开train.py文件，在如下位置修改‘–weights’，‘–cfg’，‘–data’，‘–epochs’，‘–batch-size’，‘–device’等参数的default值。
执行如下代码开始训练：

python train.py

1.2Yolov7测试

修改detect.py文件中的‘–weights’，‘–source’，‘–img-size’，‘–device’等参数的default值。

执行如下代码测试模型：

python detect.py