说明
本文主要介绍OpenPose的安装,安装环境
项目 | 版本 |
---|---|
系统 | ubuntu 16 |
cpu | 4核 |
gpu | gtx 1080 |
${OpenPose Dir}
指的是OpenPose
目录
编译时间比较长,建议在scree
中安装
获取github中最新的OpenPose代码
git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git
编译caffe
需要编译OpenPose
中的caffe
,使用自己安装的可能会执行错误
- 进入
caffe
所在目录,并创建caffe
编译输出目录
cd ${OpenPose Dir}/3rdparty/caffe
mkdir build
cd build
- 执行cmake进行编译
cmake ..
- make -j4 其中4是计算机核数
make -j4
安装GPU驱动程序
- 进入sh脚本所在路径
cd ${OpenPose Dir}/scripts/ubuntu
- 运行sh脚本
sh install_cuda.sh
获取训练原本集
如果没有获取样本集,之后运行的时候会报错
Failed to parse NetParameter file
- 进入获取样本集脚本所在目录
cd ${OpenPose Dir}/models
- 执行获取样本集脚本
sh getModels.sh
编译OpenPose
- 新建输出文件夹
cd ${OpenPose Dir}
mkdir build
cd build
- 执行cmake
cmake -DWITH_3D_RENDERER=ON -DBUILD_CAFFE=OFF -DBUILD_PYTHON=ON -DCaffe_INCLUDE_DIRS=/openpose/3rdparty/caffe/include/ -DCaffe_LIBS=/openpose/3rdparty/caffe/build/lib/libcaffe.so ..
这里需要使用OpenPose
中的Caffe
Caffe_INCLUDE_DIRS
指定了caffe
依赖
Caffe_LIBS
指定了编译后的caffe
动态链接库
BUILD_PYTHON=ON
编译python
更多参数参考github官方文档
- 执行make
make -j`nproc`
验证安装成功
我这边是服务器环境没有显示器,如果需要实时显示结果可以参考官方github中的demo命令
cd ${OpenPose Dir}
mkdir output
./build/examples/openpose/openpose.bin --video examples/media/video.avi --write_video output/result.avi --write_json output/ --display 0
显卡查看运行中的任务,可以看到显卡中有任务在跑,即安装成功
nvidia-smi