2022年最新AlphaPose环境配置(Linux+GPU)

目录

1.AlphaPose简介

2.环境配置

1.我的环境

2.准备AlphaPose的环境

3.配置AlphaPose

3.配置完成


1.AlphaPose简介

人体姿势识别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

2.环境配置

1.我的环境

先介绍一下我的环境:Ubuntu 18.04.6 LTS (GNU/Linux 5.4.0-122-generic x86_64)

我已经预先安装了anaconda、git、cuda10.1,如果没有安装的话需要安装一下。anaconda和cuda的安装可参考其他文章,git的话直接pip install git即可

2.准备AlphaPose的环境

基本需求

 

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

e5e1d20d9b2c8fc73a8b3fb2b708767f.png

因为我的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__) #注意是双下划线

877112a0b2eec42a634e3205f60c755e.png

4. 下载AlphaPose

  • 用git命令直接clone到本地
git clone https://github.com/MVIG-SJTU/AlphaPose.git

如果想下载到指定目录下,可以采用git clone 源码地址 本地路径的形式

(如果没有安装git,先用命令pip install git下载安装git)

  • 进入文件夹中cd AlphaPose

3.配置AlphaPose

  • 添加路径:
export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH
  • 下载cpython库:python -m pip install cython
  • 安装和更新:sudo apt-get install libyaml-dev
  • 注意!!!
  • for users who want to use 1.5 < PyTorch < 1.11, please switch to the pytorch<1.11 branch by:
    git checkout "pytorch<1.11"
  • for users who want to use PyTorch < 1.5, please switch to the pytorch<1.5 branch by: 
     git checkout "pytorch<1.5"
  • 以上完成后运行:python setup.py build develop

依据报错,执行相应操作。

e6a7bd24de0a701c7d7e358fd19d6262.png

如图,用pip install 库命令下载对应的库(easydict、halpecocotools、opencv-python、pyyaml、scipy、tensorboardx、terminaltables、tqdm、visdom)

特别注意,其中scipy需要指定1.1.0版本pip install scipy==1.1.0

8f2ec75782f12c94bb489c9210d0c511.png

如果出现上述错误,是由于新版本pip的特性问题,用pip install pip==21.2将pip版本切换到旧的21.2版本再次运行即可

40f3f95fb63d69bd85f097e0f12a2b55.png

如果执行完后出现如上界面,那就恭喜你,环境配置终于完成了!!

3.配置完成

本文写于2022年8月,当时网上关于AlphaPose环境配置的教程不是落后就是不够详细,没办法,只能自己一步一步去尝试着配置,最后有了这篇教程。虽然看着挺简单,但真的自己去尝试、去解决一个个报错,真的会花费很多时间,也希望这篇文章能帮到屏幕前一样正在配置AlphaPose环境的你!

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇

)">
下一篇>>