树莓派配置YOLOv7

前言

​ 笔者目前仍在学习阶段，本文章只为记录配置yolov7过程，以防日后忘记。

正文

代码克隆与环境配置

​ 首先在终端克隆一下代码。

git clone https://github.com/WongKinYiu/yolov7

​ 然后先不着急安装文件夹里面的requirement.txt，先把pytorch安装一下，否则有可能会导致pytorch版本不适配树莓派（具体我也不是很懂）

​ 进入该网站 https://download.pytorch.org/whl/torch_stable.html后，寻找自己需要的版本即可。笔者用的是cpu/torch-1.8.1-cp39-cp39-manylinux2014_aarch64.whl和cpu/torchvision-0.9.1-cp39-cp39-manylinux2014_aarch64.whl。

​ 下载好后，运行一下安装命令。然后pytorch不出意外的话就安装好啦。

pip3 install torch-1.8.1-cp39-cp39-manylinux2014_aarch64.whl
pip3 install torchvision-0.9.1-cp39-cp39-manylinux2014_aarch64.whl

​ 安装完pytorch之后，就可以去安装requirement.txt了。

cd yolov7
pip3 install -r requirement.txt

​ 然后双手离开键盘泡杯咖啡吃点零食，慢慢等待进度条就好了。也许会有几个橙色的warning，不过只要没有红色应该就基本ok。

下载权重并试运行

​ 首先先建立一个存放权重的文件夹。

mkdir weights
cd weights

​ 然后去这个网站看一下https://github.com/WongKinYiu/yolov7/releases。先下载yolov7_training.pt，这个是预训练模型，之后训练的时候得用。

wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7_training.pt

​ 再下载个已经训练好的模型，用来试运行。笔者一般会用yolov7.pt或者yolov7-tiny.pt。

wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt
wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt

​ 下载好之后，就要开始试运行了。运行以下命令（以yolov7.pt为例）。

cd ..
python detect.py --weights weights/yolov7.pt --source inference/images 

​ 这时，有可能会出现OpenBLAS Warning : Detect OpenMP Loop and this application may hang这样的报错（我也不知道算不算报错，总之就是一种非正常状态）。ctrl+c停止运行后，执行以下命令。

export OMP_NUM_THREADS=1

​ 再运行一次，应该问题就解决了。具体原因我也不知道。

训练模型

还没训练过，不会