人体姿态估计模型openpose(GPU)安装教程,超详细!Ubuntu16.04(caffe)

  由于项目原因,本人转向了人体姿态估计,查看网上浮现openpose的教程有很多错误,经过两天的时间加上和大牛交流成功安装好openpose(不把安装caffe的时间算上,因为我电脑之前已经安装好caffe,嘿嘿),下面我会详细介绍安装过程,以此让小白少走弯路。

我的电脑配置:Ubuntu16.04,NVIDIA-387,cuda8.0,cudnn5.1,opencv3.1.0

第一步:安装caffe

https://blog.csdn.net/yhaolpz/article/details/71375762

https://blog.csdn.net/yggaoeecs/article/details/79163789

个人建议:搞深度学习的人都知道安装caffe是一项必备技能,以上链接算是在配置caffe教程里的良心之作,开始配置caffe的时候我就按照上面的学习的,由于自己是个男生比较粗心,每次配置都会多少出现一些小问题,所以按照教程要细心。流程是先安装显卡驱动,在安装CUDA和cudnn,再安装opencv,最后编译caffe。每安装一个都要确认一遍是否成功!加油!

假设你安装好了显卡驱动,cuda,cudnn,opencv,那么给你一个链接按照上面的来就可以编译成功了,因为开始的那两个教程安装caffe太繁琐了。下面是链接:

https://blog.csdn.net/linhai1028/article/details/79452900

最后运行 make runtest,测试成功如下图:

第二步:重要的一步,安装openpose

官方地址:https://github.com/CMU-Perceptual-Computing-Lab/openpose

1,下载openpose

使用git下载官网的open pose或者直接去官网下载,本人习惯用git clone。都可以的

git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git

2,安装Cmake GUI

sudo apt-get install cmake-gui

3.利用Cmake GUI 下载官方给的caffemodel模型

cd models
./getModels.sh
cd ..

4,open pose填充代码

如何找到以下界面的方法:/usr/share/里面搜索Cmake就会找到一个类似三角形的图标,打开就会出现官网给的图片,下面是我的电脑出现的:

最上面的两个路径自己写,现在你下载的openpose文件夹创建一个空的build文件夹,然后点configure

默认就这两个选项,点finish,等着,安装成功会出现下面图片

这时候要注意了,之前配置好了caffe,这时候要修改路径了,修改两个地方:

caffe_INCLUDE_DIRS和caffe_LIBS,以我的为例哈,我的caffe_INCLUDE_DIRS=/home/feng/openpose/3rdparty/caffe/include

caffe_LIBS=/home/feng/openpose/3rdparty/caffe/build/lib/libcaffe.so      然后去最上面的BUILD_CAFFE的选项!点Generate,结束。

5,编译open pose

经过上面的步骤可以发现build文件里里面多了很多东西,很正常。

编译:

cd build/
sudo make

本来以为这样就没问题了,结果出现以下错误,如图:

解决方案:找到最初编译caffe的文件夹,使用命令行(生成caffe.pb.h,然后新建空文件夹移过去)

protoc src/caffe/proto/caffe.proto --cpp_out=.
mkdir include/caffe/proto
mv src/caffe/proto/caffe.pb.h include/caffe/proto

接下来编译成功了:

6,测试阶段

上面给出的官网有如何测试,我简单的介绍一下:

测试视频:

# human pose
./build/examples/openpose/openpose.bin --video examples/media/video.avi
# human pose with face and hands
./build/examples/openpose/openpose.bin --video examples/media/video.avi --face --hand

摄像头测试:

# human pose
./build/examples/openpose/openpose.bin
# human pose with face and hands
./build/examples/openpose/openpose.bin --face --hand

图片测试:

# human pose
./build/examples/openpose/openpose.bin --image_dir examples/media/
# human pose with face and hands
./build/examples/openpose/openpose.bin --image_dir examples/media/ --face --hand

由于我的examples/media文件夹里面只有图片我只对图像测试一下,举个例子:运行 /build/examples/openpose/openpose.bin --image_dir examples/media/ 出现如下错误:

看到这些错误我认为是gpu的问题,以为以前在跑目标检测模型caffe-ssd的时候设置gpu不对出现过类似问题,经过查阅发现在命令行加一个sudo就可以了,神奇。

注:这个教程是利用训练好的caffemodel测试,不包括训练模型,查阅知乎牛人说训练很吃显卡而且时间长,以后可能会尝试。

论文网址支持:https://arxiv.org/abs/1611.08050

猜你喜欢

转载自blog.csdn.net/qq_38522972/article/details/82865184