一、背景
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,然后把之前的数据集路径放进去。
- 一定