环境和上一篇安装caffe是一样的,安装openpose的难点主要是caffe的安装。。
正文。。。
依赖: Ubuntu20.04 cuda10.1 cudnn7.6.0 OpenCV4.5.1 Python 3.8+Numpy
1. 下载源码及模型
这一步下载速度可能很慢,除非你能xx,原因你们懂得。。。也可以百度一下提前下载好模型放到对应的位置
cd openpose cd models ./getModels.sh cd ..
2. 下载第三方库
因为里面的两个第三方库是空的
cd 3rdparty/ git clone https://github.com/CMU-Perceptual-Computing-Lab/caffe.git git clone https://github.com/pybind/pybind11.git
3. 手动编译caffe
这里的安装步骤和上一篇安装caffe是一样的,所以移步参考这一篇文章:
ubuntu20.04+cuda10.1+cudnn7.6.0+opencv4.5.1+python3.8安装caffe
只不过在复制Makefile.config的时候更改一下
cp Makefile.config.Ubuntu16_cuda8.example Makefile.config
编译成功后,会生成 caffe/build/lib/libcaffe.so 文件
4. cmake
打开cmake-gui界面
cmake-gui
点击configure,然后默认选择点击finish
Unix Makefiles Use default native compilers
然后更改成这样
再点击configure,然后再点击generate
5. 编译openpose
在openpose根目录下
mkdir build make -j8
如果编译过程中出现过一个错误 cannot find #include “caffe/proto/caffe.pb.h”
# 在caffe根目录 protoc src/caffe/proto/caffe.proto --cpp_out=. mkdir include/caffe/proto mv src/caffe/proto/caffe.pb.h include/caffe/proto
然后在build目录下重新编译
make clean make -j8
6. 测试
在根目录下
./build/examples/openpose/openpose.bin --video examples/media/video.avi