软硬件环境
- windows 10 64bit
- cuda 10.1
- cudnn 7.6.35
- anaconda with python 3.7
- ubuntu 18.04 64bit
- NVidia GTX 1070Ti
- cmake 3.18.4
- protobuf 3.8.0
简介
OpenPose
人体姿态识别项目是美国卡耐基梅隆大学(CMU
)基于卷积神经网络和监督学习并以caffe
为框架开发的开源库。可以实现人体动作、面部表情、手指运动等姿态估计。适用于单人和多人,具有极好的鲁棒性,是世界上首个基于深度学习的实时多人二维姿态估计应用。人体姿态估计技术在体育健身、动作采集、人机交互、舆情监测等领域具有广阔的应用前景。
算法体验
如果仅仅是想体验下这个项目,官方提供了windows
版的可执行文件下载,有gpu
和cpu
2个版本,地址是: https://github.com/CMU-Perceptual-Computing-Lab/openpose/releases/,解压后进入文件夹,下载项目中需要用到的各种模型
cd openpose\models
.\getModels.bat
使用本地视频文件进行测试
cd ..
bin\OpenPoseDemo.exe --video examples\media\video.avi
源码编译
我这里是有gpu
环境,因此需要安装nvidia
的驱动,cuda
和cuDNN
,具体安装方法前面的博文已经介绍过了,参考下面链接
- nvidia驱动安装,https://xugaoxiang.com/2020/09/24/ubuntu-nvidia-driver/
- CUDA和cuDNN安装, https://xugaoxiang.com/2019/12/13/ubuntu-cuda/
git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git
cd openpose
conda create -n openpose python=3.7
conda activate openpose
sudo apt install cmake build-essential
mkdir build
cd build
cmake ..
make -j `nproc`
cd ../models
./getModels.sh
cd ..
./build/example/openpose.bin --video examples/media/video.avi
如果原来的系统中已经安装过了caffe
,那么openpose
可以直接使用而不需要再次安装了
cmake -DBUILD_CAFFE=OFF -DCaffe_INCLUDE