目录
1.AlphaPose简介
2.环境配置
1.我的环境
2.准备AlphaPose的环境
3.配置AlphaPose
3.配置完成
人体姿势识别AlphaPose是一款精准的多人姿态评估工具,是上海交通大学卢策吾团队开源的作品。
AlphaPose is an accurate multi-person pose estimator, which is the first open-source system that achieves 70+ mAP (75 mAP) on COCO dataset and 80+ mAP (82.1 mAP) on MPII dataset. To match poses that correspond to the same person across frames, we also provide an efficient online pose tracker called Pose Flow. It is the first open-source online pose tracker that achieves both 60+ mAP (66.5 mAP) and 50+ MOTA (58.3 MOTA) on PoseTrack Challenge dataset.
AlphaPose supports both Linux and Windows!
具体信息可前往GitHub查看GitHub - MVIG-SJTU/AlphaPose: Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System
先介绍一下我的环境:Ubuntu 18.04.6 LTS (GNU/Linux 5.4.0-122-generic x86_64)
我已经预先安装了anaconda、git、cuda10.1,如果没有安装的话需要安装一下。anaconda和cuda的安装可参考其他文章,git的话直接pip install git
即可
基本需求
1. 首先先创建一个用于AlphaPose的虚拟环境:conda create -n AlphaPose python=3.7
(Python的版本号可以修改,看个人需求)
如果想删除环境的话可以用:conda remove -n AlphaPose –all
2. 激活并进入虚拟环境:conda activate AlphaPose
3. 安装pytorch,去pytorch官网复制对应cuda版本的pytorch安装命令即可
Previous PyTorch Versions | PyTorch
因为我的cuda版本为10.1,所以选择了1.6.0版本的pytorch,命令为:
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
(如果下载速度慢,可以切换成国内镜像源下载,这里不赘述,自行查阅资料)
下载完成后,可以进入python终端,查看是否下载成功
python
import torch
print(torch.__version__) #注意是双下划线
4. 下载AlphaPose
git clone https://github.com/MVIG-SJTU/AlphaPose.git
如果想下载到指定目录下,可以采用git clone 源码地址 本地路径
的形式
(如果没有安装git,先用命令pip install git
下载安装git)
cd AlphaPose
export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH
python -m pip install cython
sudo apt-get install libyaml-dev
git checkout "pytorch<1.11"
git checkout "pytorch<1.5"
python setup.py build develop
依据报错,执行相应操作。
如图,用pip install 库
命令下载对应的库(easydict、halpecocotools、opencv-python、pyyaml、scipy、tensorboardx、terminaltables、tqdm、visdom)
特别注意,其中scipy需要指定1.1.0版本pip install scipy==1.1.0
如果出现上述错误,是由于新版本pip的特性问题,用pip install pip==21.2
将pip版本切换到旧的21.2版本再次运行即可
如果执行完后出现如上界面,那就恭喜你,环境配置终于完成了!!
本文写于2022年8月,当时网上关于AlphaPose环境配置的教程不是落后就是不够详细,没办法,只能自己一步一步去尝试着配置,最后有了这篇教程。虽然看着挺简单,但真的自己去尝试、去解决一个个报错,真的会花费很多时间,也希望这篇文章能帮到屏幕前一样正在配置AlphaPose环境的你!