Windows10下的AlphaPose配置教程

发表于

介绍

AlphaPose是一个精确的多人姿态估计器，它在COCO、MPII等数据集上效果显著高于其他已开源系统，而且同时支持Linux和Windows，并且是交大MVIG组开发的，所以来尝试一下。另外AlphaPose还提供了基于PyTorch平台的源码，可以说很方便使用了，并且安装教程也很详细。

配置环境

  1. 安装Git: https://git-scm.com/download/win

  2. 安装Python3.6 or 3.7: https://www.python.org/downloads/

  3. 安装CUDA 9.0 （必须是受支持的N卡） : https://developer.nvidia.com/cuda-90-download-archive

  4. 根据Python、CUDA版本安装PyTorch 0.4.1: https://pytorch.org/get-started/previous-versions/

    在Git Bash中输如下命令即可：

    1
    2
    pip3 install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
    pip3 install torchvision

    如果pip下载过慢可以先用IDM等下载工具自己下载whl文件然后用pip安装本地的whl：

    Python3.6：https://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl

    Python3.7：https://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-win_amd64.whl

  5. 下载最新的win-64/intel-openmp bz2文件 https://anaconda.org/anaconda/intel-openmp/files 把压缩包中Library\bin目录下的dll文件放到一个电脑的某个文件夹中，然后将这个文件夹加入环境变量。（推荐放到C:\Users\Your user name\bin）

  6. 按照官方的安装方法git clone PyTorch版本项目，然后pip install -r requirements.txt即可

    如果提示torch版本不对，就把requirements.txt中的torch版本号改成你当前使用的版本号。

  7. 手动下载Model文件: duc_se.pth (2018/08/30) (Google Drive | Baidu pan), yolov3-spp.weights(Google Drive | Baidu pan). 分别放到 ./models/sppe 和 ./models/yolo 文件夹下

使用

以使用webcam为例：

1
python3 webcam_demo.py --vis

使用 demo.py 和 video_demo.py 时需要开启 --sp 参数，比如：

1
python3 demo.py --indir ${img_directory} --outdir examples/res --sp

结果

识别结果还可以

3.jpg