search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
1.91k stars 311 forks source link

ADD multiview human pose #294

Open kyakuno opened 3 years ago

kyakuno commented 3 years ago

https://github.com/microsoft/multiview-human-pose-estimation-pytorch MIT

hattan1990 commented 3 years ago

1.公式データセット（MPII+H36M）をMPIIデータのみに変更 　ダウンロードする際のアカウント登録が学術研究機関などのドメインに制限されており、H36Mデータの取得が困難なため 　※精度改善が必要な場合、H36Mデータの取得について要検討

2.MPIIデータを用いて、学習を実施（パラメータ：256_nofusion.yaml） 　python run/pose2d/train.py --cfg experiments-local/mpii/mobilenetv2/256_nofusion.yaml 　※backboneモデルは、mobilenetv2を使用

3.モデル検証を実施して、3D推論に必要な「predicted_heatmaps.h5」を取得 　python run/pose2d/valid.py --cfg experiments-local/mpii/mobilenetv2/256_nofusion.yaml

hattan1990 commented 3 years ago

補足：「predicted_heatmaps.h5」は、modelの推論から生成可能なため、h5ファイルとして保存する必要はない