yolov7安装/推理/训练操作记录

这一部分参考网址。

官方YOLOv7 项目地址：https://github.com/WongKinYiu/yolov7。下载main分支下的文件，下载完成后把yolov7-main文件夹下的文件转移到第一步新建的python项目文件夹下（如果直接将yolov7-main文件夹放在项目文件夹下，训练时的文件相对路径会比较麻烦）。

接下来开始安装yolov7运行所需要的python库以及一些.pt文件。目前官方YOLOv7 项目自带requirements.txt文件，打开项目文件，在终端中输入：

yolov7运行所需的库就安装完毕。在官方项目中提供了一些训练好的权重文件（.pt文件），可以在页面另行下载。我主要用的是yolov7-tiny版本，官网项目中似乎并未提供yolov7-tiny.pt，需要自己找一找。

以上步骤完成后，可以先尝试做一次推理，验证项目能否运行。

打开detect.py文件夹，在最下面的参数设置中将权重文件改为自己项目中权重文件的路径：

这里我在项目中新建了weights文件夹，把权重文件放在该文件夹中，所以路径改为'weights/yolov7.pt'。

可以在代码中修改参数，也可以在Terminal输入指令指定参数，见网址。

推理用的默认图片在inference/images文件夹中，推理的结果放在runs/detect/exp文件夹中（1650显卡推理时得不到预测框，具体情形在报错记录里展示）。

训练过程参考网址。我把数据集放在了与项目文件同级的文件夹中（似乎是yolov5的要求），不知道yolov7能不能放在其他位置。

训练得到的结果存在runs/train/exp文件夹中，包括初始权重文件/最后权重文件/最好权重文件（best.pt）。

解决办法：将detect.py中的half参数设置为False（第31行改为：half = False）。

Traceback (most recent call last): File "D:\...\yolov7\train.py", line 616, in train(hyp, opt, device, tb_writer) File "D:\...\yolov7\train.py", line 86, in train attempt_download(weights) # download if not found locally File "D:\Files\TensorFlow\Yolov7\yolov7\utils\google_utils.py", line 31, in attempt_download tag = subprocess.check_output('git tag', shell=True).decode().split()[-1]

我已经有yolov7.pt文件，这里的问题时文件相对位置有误，参考第二节的说法改正。

应该是训练时分配的内存不足。有两种解决办法：