AlphaPose环境配置与测试

1. AlphaPose环境配置

项目地址：https://github.com/MVIG-SJTU/AlphaPose
（1）安装anaconda（参考链接1.1）https://blog.csdn.net/weixin_50008473/article/details/115250986?spm=1001.2014.3001.5501
（2）创建虚拟环境：conda create -n alphapose python=3.6 -y
（3）进入虚拟环境：conda activate alphapose
（4）安装pytorch：conda install pytorch==1.1.0 torchvision==0.3.0
若安装失败可多试几次，或者下载太慢可以换源（https://blog.csdn.net/weixin_50008473/article/details/116485755?spm=1001.2014.3001.5501）
（5）安装相应依赖：

export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH

pip install opencv-python==4.1.0.25
pip install cython
pip install tqdm
pip install natsort
pip install pycocotools

sudo apt install git
sudo apt-get install libyaml-dev

python setup.py build develop

安装成功后如下图

若最后一步出错可以多试一次

2. AlphaPose算法测试 2.1 准备模型

百度云下载链接: https://pan.baidu.com/s/1QW7UVJe1O1zxsfyKqBJo6w
提取码: e3pe

1.下载行人检测模型：yolov3-spp.weights；放到detector/yolo/data文件夹下
2.下载姿态关键点检测模型（如下图所示任选一个，注意：不同的model对应不同的cfg），放到AlphaPose/pretrained_models文件夹下
https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/MODEL_ZOO.md

3.下载行人重识别模型：osnet_ain_x1_0_msmt17_256x128_amsgrad_ep50_lr0.0015_coslr_b64_fb10_softmax_labsmth_flip_jitter ，放到trackers/weights/文件夹下

2.2 测试

1.测试examples/demo文件夹下的图片

python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_2x-dcn.yaml --checkpoint pretrained_models/fast_dcn_res50_256x192.pth --indir examples/demo/ --vis --showbox --save_img

结果保存在examples/res/与examples/res/vis/目录下

跟踪的话，在上面命令基础上加–pose_track

python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_2x-dcn.yaml --checkpoint pretrained_models/fast_dcn_res50_256x192.pth --indir examples/demo/ --vis --showbox --save_img --pose_track

如果使用了很多图片或者一个视频，需要加上–sp参数

python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_2x-dcn.yaml --checkpoint pretrained_models/fast_dcn_res50_256x192.pth --indir examples/demo/ --vis --showbox --save_img --pose_track --sp

QA小结：
1.ImportError : cannot import name ‘deform_conv_cuda’
方法：打开alphapose/setup.py，把第124行：force_compile = False改为True

执行：

rm -rf build
python setup.py build develop

2.ValueError:source code string cannot contain null bytes
原因1：导入模块中出现了多余的空字符，在 vim 里就显示成 ^@
解决方法：打开vi替换

:%s/\%x00//g

原因2：新建的__init__.py文件存在多余的空字符，改为空文件即可。