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文件存在多余的空字符,改为空文件即可。

猜你喜欢

转载自blog.csdn.net/weixin_50008473/article/details/123184361