AlphaPose是一款精準的多人姿態評估工具,並聲稱是第一款實時的開源系統。
工具支持TensorFlow和PyTorch兩種實現。AlphaPose利用一種區域性的多人動作估計框架將不精準的人類邊界框該進程精確的動作估計。這裡有三種元素:對稱空間轉換網絡(SSTN)、參數化姿態非極大抑制(NMS)以及姿態導向的生成器(PGPG)。
依賴:
Torch
TensorFlow(verson >= 1.2)
安裝:
1.下載
git clone https://github.com/MVIG-SJTU/AlphaPose.git
cd AlphaPose/human-detection/lib/
make clean
make
cd newnms/
make
cd ../../../
2.安裝Torch/TensorFlow
chmod +x install.sh
./install.sh
3.運行fetch_models.sh,下載預訓練模型。
chmod +x fetch_models.sh
./fetch_models.sh
快速開始;
運行demo,
./run.sh --indir examples/demo/ --outdir examples/results/ --vis
AlphaPose既可以在圖片、視頻或多圖中進行姿態估計,也能在畫面中對動作進行追蹤。它的輸出形式非常廣泛,包括PNG、JPG和AVI等具有關鍵點的圖片形式,也有JSON格式的輸出,這一特點也使其成為眾多應用受歡迎的工具。
閱讀更多 AI踐行者 的文章