YOLOV7跑通demo与train.py

  1. 下载yolov7（github）与testing里面的yolov7.pt（预训练权重文件）文件

2.新建项目，选中解压文件yolov7 

3.新建虚拟环境：

file- setting - Project yolov7- Project interpreter - add- \venv - make available 

如不出现虚拟环境可重启试试

4.

若清华源太慢，可改换豆瓣源：

pip3 install -r requirements.txt  -i https://pypi.douban.com/simple

 5.

导入yolov7.pt文件（其它文件也可但需要在default修改）

6.run detect.py

运行train.py(跑自己的训练集):

1.制作数据集

数据集- images-train/val

             labels -train/val(标签内有 分类 中心点的坐标 目标框的宽长)

             train/val.text--图片路径（D:/yolov7/data/images/0001.jpg）

2.

 workers---加载图片的数量（cpu最好修改一下） device---设备

 batch-size --每次训练批次的大小

data--数据集的位置

 

img--图片大小

cfg cfg/training/yolov7.yaml 

default = 'yolov7.pt'

运行yolov7--train.py