在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!