一.下载源码

在github中下载Yolov7源码并解压缩
YOLOv7 源码

数据标注可以参考这篇文章
https://blog.csdn.net/qq_39779449/article/details/110172948?spm=1001.2014.3001.5502

二.配置环境

整体配置树结构

安装Yolo v7 gpu 环境

在Anconda中创造Yolov7环境

安装所需各种包

激活Yolov7环境 并安装Yolov7所需各种包，采用清华镜像源安装所需包（都试了一下，还是清华镜像更快）

activate yolov7

pip install -r F:\objectdet\yolov7-main/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

检查刚刚环境安装是否成功


如果完全按照requirements.txt来安装的话，你会发现安装的torch版本并不是gpu版本，所以得重新安装gpu版本的才能用你的gpu来训练Yolov7

pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html -i  https://pypi.tuna.tsinghua.edu.cn/simple

安装好后，通过pip list查看你所安装的包，如下图所示，恭喜，安装成功

三.使Yolo v7 训练并测试自己的数据集

修改Yolo v7配置

在pycharm中选择刚刚安装好的Yolov7所需环境
修改配置环境

新建自己数据集的yaml文件

• train: 修改为自己的训练集路径
• val: 修改为自己的验证集路径
• test: 修改为自己的测试集路径
• nc: 修改为自己的数据集类别数量
• names: 修改为自己的数据集类别标签。