YOLOv7环境搭建、训练流程以及转TensorRT部署问题

一、背景

github官网yolov7，代码什么的从这个网站下，还有一个，但是这是官网

二、环境搭建

有两种环境搭建方式，一是用conda搭个虚拟环境，然后安装所有需要的库跟依赖等；二是用docker容器，下载英伟达的pytorch image镜像，里面已经装好了一部分需要的包，如torch等。建议用第二种。

1）.conda虚拟环境

conda create -n yolov7python=3.8
conda activate yolo7

2）.利用pytorch镜像创建环境

到这个网址去下载NCG catalogpytorch的镜像，比如：

docker pull nvcr.io/nvidia/pytorch:22.02-py3

然后根据该镜像起一个容器container，因为之前跑过yolov5，所以直接用yolov5的环境就可以

三、数据集

• 这是想说的重点！官网要求用coco数据集，可以将自己数据集转成coco格式使用。但是其实并不必要，用之前的yolov5的数据集就可以。
• 根据data文件中的coco.yaml写一个自己的yaml，然后把之前的数据集路径放进去。<