原文鏈接：https://blog.techbridge.cc/2019/01/18/openpose-installation/

注：轉(zhuǎn)載如有侵權(quán)，請(qǐng)馬上聯(lián)系刪除。

如何在 Windows 安裝 OpenPose 跟使用 Python API 來(lái)偵測(cè)人體姿態(tài)

?

前言

?

OpenPose?是 Carnegie Mellon University（CMU）論文實(shí)作的開(kāi)源函式庫(kù)，提供的功能主要就是可以偵測(cè)人體的各部位在什麼地方，例如可以在影像中找到臉、身體、手等等地方的特徵點(diǎn)。但他厲害的地方在於可以一次偵測(cè)很多人，看看下面這張圖就知道

這個(gè) repository 仍然相當(dāng)?shù)?active，有兩個(gè) CMU 的學(xué)生常常在 Github 上面解 issue。這次因?yàn)楣P者需要在 Windows 上面安裝 OpenPose 並呼叫 Python API，所以跟開(kāi)發(fā)者有不少互動(dòng)，才想到乾脆來(lái)寫(xiě)一篇教學(xué)，讓之後想要使用 OpenPose 的讀者可以省去好幾天的環(huán)境安裝 debug。

OpenPose 演算法簡(jiǎn)介

上面是 OpenPose 的架構(gòu)圖，輸入會(huì)是一張大小為 w x h 的彩色影像；接著會(huì)先經(jīng)過(guò) Stage 0 的 VGG-19 Network，得到輸入影像的 feature map；然後會(huì)在 Stage 1 經(jīng)過(guò)一些 CNN 來(lái)得到身體各部位的 confidence map。Stage 1 分成兩個(gè) branch，branch 1 可以產(chǎn)生某特定部位的 confidence map，例如左肩的 confidence map：

branch 2 則是產(chǎn)生不同身體部位之間的關(guān)聯(lián)性，例如脖子跟左肩的關(guān)係：

最後結(jié)合這些結(jié)果就可以得到全身各部位大致在什麼地方，還有彼此相對(duì)位置的關(guān)係。以上的講法有點(diǎn)過(guò)於簡(jiǎn)化，有興趣的讀者可以看看延伸閱讀 1 的原始論文。

Windows 安裝步驟

1. 確認(rèn)一下你的硬體跟作業(yè)系統(tǒng)符合?OpenPose 要求

2. Clone OpenPose 的原始碼

   因?yàn)槲覀円褂?Python API，所以不能只是下載已經(jīng)編譯好的 library 文件，必須要從原始碼開(kāi)始編譯。我推薦大家可以用 Github Desktop 來(lái)下載原始碼 & 管理。

3. 下載並安裝 CMake GUI

   可以上?CMake 的網(wǎng)站下載，Windows 的話(huà)請(qǐng)下載? cmake-X.X.X-win64-x64.msi 。

4. 安裝 Visual Studio 2015

   OpenPose 的官方要求是 Visual Studio 2015 Enterprise Update 3，但似乎也有人用 Visual Studio 2015 Community 安裝成功。筆者是用 Enterprise 版本安裝成功的。

5. 安裝 CUDA

   官方推薦的版本是 CUDA 8。這一步要在安裝完 Visual Studio 之後做，因?yàn)榘惭b過(guò)程會(huì)產(chǎn)生一些 Visual Studio 需要的檔案。

6. 安裝 cuDNN

   官方建議安裝 5.1 版本。安裝方法很簡(jiǎn)單，只要把下載的壓縮檔內(nèi)容解壓縮到? C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ?路徑就好。（cuDNN 壓縮檔裡面會(huì)有三個(gè)資料夾，可以對(duì)應(yīng)到? C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 ?裡面也有的資料夾，分別把檔案放到對(duì)應(yīng)的資料夾就好）。

7. 開(kāi)始設(shè)定 CMake 來(lái)準(zhǔn)備編譯需要的檔案

   • 設(shè)定檔案路徑（在 clone 下來(lái)的路徑中自己創(chuàng)一個(gè) build 資料夾，並設(shè)定成? Where to build the binaries ）
     

   • 按下 Configure 按鈕（generator 記得選 Visual Studio 14 2015 Win64）
     
     

   • 等待 Configure Done

   • 勾選 BUILD_PYTHON

   • 按下 Generate

8. 用 Visual Studio 2015 打開(kāi)? build/OpenPose.sln ?檔案

9. 切換到 Release Mode 並 Build Project

這一步很重要，因?yàn)橹挥?Debug Mode build 會(huì)讓後面的 Python API 啟動(dòng)失敗。詳見(jiàn)?issue 1026。

Python API 呼叫步驟

如果上面的步驟都做完，應(yīng)該可以在? openpose\build\python\openpose\Release ?看到? openpose_python.cp36-win_amd64.pyd ?library 文件。（OpenPose 原生是用 C++ 寫(xiě)的，是用 pybind11 包成 Python 可以呼叫的 library）。

接著，我們準(zhǔn)備要來(lái)跑? openpose\build\examples\tutorial_api_python\1_body_from_image.py ，要記得把裡面的 library 搜尋路徑改成自己的：

我可以跑起來(lái)的範(fàn)例如下：

                    
                      # Import Openpose (Windows/Ubuntu/OSX)
dir_path = os.path.dirname(os.path.realpath(__file__))
try:
# Windows Import
if platform == "win32":
# Change these variables to point to the correct folder (Release/x64 etc.)
sys.path.append(dir_path + '/../../python/openpose/Release');
os.environ['PATH'] = os.environ['PATH'] + ';' + dir_path + '/../../x64/Release;' + dir_path + '/../../bin;'
import openpose_python as op

                    
                  

這時(shí)，你就可以去 cmd.exe 執(zhí)行，執(zhí)行下列步驟

                    
                      ::切換到你自己的 openpose 路徑
cd openpose\build\examples\tutorial_api_python
python 1_body_from_image.py

                    
                  

然後跑出下面的結(jié)果：

總結(jié)

今天跟大家介紹了 CMU 的 OpenPose 要怎麼安裝，也稍微介紹了 Python API 要怎麼使用，理論上學(xué)完這篇教學(xué)的內(nèi)容後，你就可以用 Python 呼叫 OpenPose 的 API 來(lái)實(shí)作自己想要的更高階功能了。希望對(duì)你的研究或 project 有幫助！

延伸閱讀

1. Convolutional Pose Machines

關(guān)於作者：
@pojenlai?演算法工程師，對(duì)機(jī)器人、電腦視覺(jué)和人工智慧有少許研究，正在學(xué)習(xí)用心體會(huì)事物的本質(zhì)跟不斷進(jìn)入學(xué)生心態(tài)改進(jìn)。

喜歡我們的文章嗎？歡迎分享按讚給予我們支持和鼓勵(lì)！