YOLOv7-Pose 姿态估计-环境搭建和推理

目录

1、Yolov7-pose源码下载

2、环境搭建

3、模型推理

---

1、Yolov7-pose源码下载

下载地址：GitHub - WongKinYiu/yolov7 at pose

 Yolov7-pose权重下载

下载好后打开yolov7-pose源码包

2、环境搭建

打开Anaconda3终端，进入base环境

创建新环境，我这里创建的是p38t17(python3.8，pytorch1.7)

conda create --name p38t17 python==3.8

进入到创建的环境

conda activate p38t17

安装pytorch：（网络环境比较差时，耗时会比较长）

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i  https://pypi.tuna.tsinghua.edu.cn/simple/

查看torch是否安装成功

conda list

打开工程后，进入设置

选择环境

打开终端

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3、模型推理

打开detect.py，我做了如下修改：

modelpath:模型的存放路径

imgpath：需要预测的图片的存放路径

运行报错如下：

修改如下：

运行：

运行结果保存在runs文件夹下