YOLOv7——训练自己的数据集

论文地址：https://arxiv.org/abs/2207.02696

源码地址：https://github.com/WongKinYiu/yolov7

下载好代码包，解压后配置环境，在终端直接下载requirements.txt的代码就好

（本人环境：torch 1.8.0，当然有一些tensorboard、wandb等工具包需要自己下载啦~）

pip install -r requirements.txt

wandb安装教程看这个：wandb使用_ai-ai360的博客-CSDN博客_wandb使用

输入后就是以下情况（我的环境名称叫做yolov5，这个忽视掉....）

一、数据集准备

本人数据集是自己标注的VOC格式的建筑裂缝目标检测数据集，以此为例。

（✔文件中的内容需要自己提前准备好的， ImageSets - Main、labels、yolo_path是需要自己创建的空文件夹，文件里的内容后续代码生成。）

文件分别为：✔Annotations（.xml格式标注标签）

                      ImageSets - Main（存放训练集、测试集等划分.txt格式文件）

                      ✔JPEGImages（.jpg格式图像）

                      labels（转化为yolo可使用的.txt格式标注标签）

                      yolo_path（训练集、测试集等.txt格式路径&#x