一、环境配置
1.快捷键win+r,输入cmd进入命令行窗口,使用nvcc -V
命令查看自己的cuda版本
- 我的是cuda11.1版本
- cuda的安装教程去百度安装教程即可,很简单的😁
2.创建一个conda环境
create -n Alphapose python=3.7
3.进入这个环境中
conda activate Alphapose
4.去pytorch官网(地址)找到符合自己pytorch版本的命令
我的安装命令如下,仅供参考哈:
`pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html`
**注意pytorch版本一定要和cuda版本对应。
二、Alphapose相关**
1.从这个地址 https://github.com/Amanbhandula/AlphaPose将alphapose代码下载到本地
2.从github中下载两个模型duc_se.pth和Yolov3-spp.weights分别放入models/sppe文件夹和models/yolo文件夹中,权重文件的百度云链接在项目主页如下位置:
3.从上面的环境中进入到下载项目地址
4.安装requirements.txt中所要求的依赖项。注意:有些依赖是不用安装的,直接去掉;还有 opencv-python这个包需要手动安装。
下图是修改后的:
执行 pip install -r requirements.txt
opencv-python建议安装这个版本 pip install opencv-python==4.1.0.25
如果还是有问题仍需降低版本到3点几版本。
5.在运行时缺少包直接pip install 按照提示安装即可。
三、代码运行
1.使用pycharm打开项目,更改环境为第一步配置的环境,这步就不多说了。
2.本地视频测试:从README.md中复制命令
到下图这个位置,记得将地址换成你本地视频地址,例如:D:\video\01.mp4
最后直接点run运行即可
这是运行成功后的图:
2.生成的视频还有json文件保存到这个目录下
AlphaPose_01