Ubuntu16.04从源码安装DensePose

Installing DensePose

DensePose官网 http://densepose.org/

DensePose代码 https://github.com/facebookresearch/Densepose

DensePose安装系统要求：有NVIDIA GPU的Linux系统，满足这一条件便可按照下面的步骤安装。

安装Anaconda

官网下载Linux版的Anaconda3，运行bash Anaconda3-5.2.0-Linux-x86_64.sh。

安装caffe2(pytorch)

创建caffe2的conda环境

conda create -n densepose python=2.7

进入新创建的环境

source activate densepose

安装一些需要的库

(densepose)$ conda install -y future gflags leveldb mkl mkl-include numpy opencv protobuf
(densepose)$ conda install flask graphviz hypothesis jupyter matplotlib pydot pyyaml requests scikit-image scipy setuptools tornado

下载pytorch及其依赖的第三方库

# git clone --recursive https://github.com/pytorch/pytorch.git 
(densepose)$ git clone -b v0.4.1 --recursive https://github.com/pytorch/pytorch.git # 下载v0.4.1分支代码
(densepose)$ cd pytorch
(densepose)$ git submodule update --init

编译并安装

(densepose)$ rm -rf build && mkdir build && cd build

(densepose)$ cmake -DCMAKE_PREFIX_PATH=~/anaconda3/envs/densepose -DCMAKE_INSTALL_PREFIX=~/anaconda3/envs/densepose -DUSE_NATIVE_ARCH=ON ..

(densepose)$ make -j32 install

测试是否安装成功

# To check if Caffe2 build was successful
(densepose)$ python2 -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
# 输出Success即安装成功,否则失败

# To check if Caffe2 GPU build was successful
(densepose)$ python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'
# 输出8或者其他,表示显卡个数

安装cocoapi

# COCOAPI=/path/to/clone/cocoapi
(densepose)$ git clone https://github.com/cocodataset/cocoapi.git $COCOAPI
(densepose)$ cd $COCOAPI/PythonAPI
# Install into global site-packages
(densepose)$ make install
# Alternatively, if you do not have permissions or prefer
# not to install the COCO API into global site-packages
(densepose)$ python2 setup.py install --user

安装densepose

(densepose)$ mkdir $DENSEPOSE_DIR && cd $DENSEPOSE_DIR
(densepose)$ git clone https://github.com/facebookresearch/densepose
(densepose)$ cd densepose
(densepose)$ pip install -r requirements.txt  # install Python dependencies

(densepose)$ make  # set up python modules
(densepose)$ python2 detectron/tests/test_spatial_narrow_as_op.py  # check that Detectron tests pass

(densepose)$ make ops  # Build the custom operators library
(densepose)$ python2 detectron/tests/test_zero_even_op.py  # check that the custom operator tests pass

下载densepose数据:

(densepose)$ cd DensePoseData
(densepose)$ bash get_densepose_uv.sh

如果需要训练,下载DensePose-COCO数据集:

(densepose)$ bash get_DensePose_COCO.sh

如果评价(evaluation),还需下载其他需要的文件:

(densepose)$ bash get_eval_data.sh

Inference with Pretrained Models

首先从这里下载与DensePose_ResNet101_FPN_s1x-e2e.yaml相匹配的训练好的模型,并放在weights文件夹下,然后执行下面命令:

(densepose)$ python2 tools/infer_simple.py 
    --cfg configs/DensePose_ResNet101_FPN_s1x-e2e.yaml 
    --output-dir DensePoseData/infer_out/ 
    --image-ext jpg 
    --wts  weights/DensePose_ResNet101_FPN_s1x-e2e.pkl 
    DensePoseData/demo_data/demo_im.jpg