在Ubuntu上安装OpenPose,你需要遵循以下步骤:
打开终端并运行以下命令来更新系统和安装必要的依赖项:
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential cmake git pkg-config
如果你有NVIDIA GPU并希望利用CUDA加速,你需要安装CUDA。请访问NVIDIA CUDA工具包下载并安装合适的版本。安装完成后,设置环境变量:
export PATH=/usr/local/cuda-<version_number>/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-<version_number>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
将<version_number>
替换为你安装的CUDA版本号。
如果你已经安装了CUDA,你还可以安装cuDNN库来进一步提高性能。请访问NVIDIA cuDNN库下载并安装合适的版本。安装完成后,设置环境变量:
export LD_LIBRARY_PATH=/usr/local/cuda-<version_number>/lib64:$LD_LIBRARY_PATH
将<version_number>
替换为你安装的CUDA版本号。
OpenPose依赖于OpenCV库。运行以下命令来安装OpenCV:
sudo apt-get install libopencv-dev
在你喜欢的目录中克隆OpenPose的GitHub仓库:
git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git
cd openpose
现在你可以编译OpenPose。运行以下命令:
mkdir build
cd build
cmake ..
make -j$(nproc)
这将使用所有可用的CPU内核进行编译,以加快编译速度。编译完成后,你将在build
目录中看到生成的可执行文件。
要运行OpenPose示例,首先下载模型文件:
wget https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/models/pose/mpi/pose_iter_160000.caffemodel?raw=true -O pose/mpi/pose_iter_160000.caffemodel
wget https://github.com/CMU-Perceptual-Computing-Lab/openpose/blob/master/models/pose/mpi/pose_deploy_linevec_faster_4_stages.prototxt?raw=true -O pose/mpi/pose_deploy_linevec_faster_4_stages.prototxt
然后运行以下命令来测试OpenPose:
./bin/OpenPoseDemo.exe --image_dir ../examples/media/ --display 0 --render_pose 0
这将在../examples/media/
目录中的图像上运行OpenPose,并将结果保存到output/
目录中。你可以通过修改命令行参数来自定义输出和其他设置。要查看所有可用选项,请运行./bin/OpenPoseDemo.exe --help
。
现在你已经在Ubuntu上成功安装了OpenPose!