SangbumChoi / MobileHumanPose

This repo is official PyTorch implementation of MobileHumanPose: Toward real-time 3D human pose estimation in mobile devices(CVPRW 2021).
MIT License
232 stars 29 forks source link

do you used 2d keypoint model? #4

Closed wangm-word closed 3 years ago

wangm-word commented 3 years ago

thank you for your work! how to use the demo code ? do you used 2d keypoints when you train and vis?

SangbumChoi commented 3 years ago

https://github.com/SangbumChoi/MobileHumanPose/issues/2

For the question of using 2d keypoints please refer to the above issue

For the demo code you might try

python3 demo.py --gpu 0-1 --modelpath {model_path} --input_image {input_image.jpg} --backbone LPSKI

However, I found that current demo code is quite old so it requires to give input of headnet but I need to fix this.

wangm-word commented 3 years ago

2

For the question of using 2d keypoints please refer to the above issue

For the demo code you might try

python3 demo.py --gpu 0-1 --modelpath {model_path} --input_image {input_image.jpg} --backbone LPSKI

However, I found that current demo code is quite old so it requires to give input of headnet but I need to fix this.

thank you very much

SangbumChoi commented 3 years ago

@wangm-word I also updated demo.py but not sure if it works :( Feel free to test