PyTorchの「3D human pose estimation」を試してみる

github.com

環境

Windows10 Pro
GPUなし
Python 3.8.2

バージョン確認（pip freeze）

インストールしたのは「torch」「torchvision」「opencv-python」「matplotlib」のみ。

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install opencv-python
pip install matplotlib

その他は勝手についてくる。

cycler==0.10.0
future==0.18.2
kiwisolver==1.2.0
matplotlib==3.2.1
numpy==1.18.5
opencv-python==4.2.0.34
Pillow==7.1.2
progress==1.5
pyparsing==2.4.7
python-dateutil==2.8.1
six==1.15.0
torch==1.5.0+cpu
torchvision==0.6.0+cpu

学習済みモデルのダウンロード

GitHubに示されている通りにGoogle Driveからダウンロードしてくる。

実行

用意されたサンプル画像

python ./src/demo.py --demo ./images/mpi_inf_3dhp_706.png --gpus -1 --load_model ./models/fusion_3d_var.pth

結果

2つの画像が出力される。
2番目の画像はグリグリ回転させることができる。