目录
1 Alphapose
github地址:https://github.com/MVIG-SJTU/AlphaPose
2 环境配置
我的电脑的环境:
Window10
Cuda11.2(因为我的显卡是3060的)
Python 3.7
torch 1.9.0
torchvision 0.10.0
2.1 创建anacoda虚拟环境
Anacoda Prompt命令行创建
conda create -n Alpha python=3.7
命令行里激活 activate Alpha
2.2 获取Alphapose
使用Git把工程down下来
git clone https://github.com/MVIG-SJTU/AlphaPose.git
此处还合并了一下目录,忘了从哪个博客里抄的了…,不合并应该也行吧…
2.3 环境配置
1、命令行里激活 activate Alpha
2、安装pytorch,指令从Pytorch官网找
3、安装cpython
pip install cython
4、运行python setup.py build develop --user
python setup.py build develop --user
可别用requirements.txt装,根本搞不成的,呜呜呜。
PS.(1)此处运行要在Alphapose工程下打开cmd,激活环境再运行。
大的坑要来了,一般来说会安装完,但假如你的电脑没装过VS,会有一个库编译不通过,此处参照了链接: Alphapose - Windows下Alphapose(Pytorch 1.1+)版本2021最新环境配置步骤以及踩坑说明.中2.5.1中的方法,但我最初没有装VS,失败,装了VS2019(安装界面选python,右边勾选python本地开发工具什么的,记不清了)后,成功编译通过。此处要从Github上把pycocotools down下来,再此文件的子目录PythonAPI下,命令行执行
python setup.py build
python setup.py install
再运行 python setup.py build develop --user,直到所有库安装完成。
(2)deform_conv_cuda,此处也参照了链接: Alphapose - Windows下Alphapose(Pytorch 1.1+)版本2021最新环境配置步骤以及踩坑说明.中2.5.3中的方法。注释掉了两行,不用DCN模型了。
3 权重下载
1、下载目标检测模型yolov3-spp.weights放到目录detector/yolo/data文件夹下
2、下载姿态模型,放置到pretrained_models文件夹下
从别的博客里找的网盘链接,随便下的
4 demo试运行
python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir examples/demo/
(其实就是readme里给的,我连路径都没改,只换了demo文件夹里的图)
最后会在res文件下生成.json文件
视频没测试,环境装了两天,孩子人快装傻了,姑且算是装好了⑧