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

Will pre-trained model be provided? #30

Closed ALISCIFP closed 1 year ago

GKeppler commented 1 year ago

Yes, please provide checkpoints. It is great for other researchers and it also provides more evidence for your results.

SangbumChoi commented 1 year ago

@ALISCIFP @GKeppler I never thought about that sorry for inconvinience. I will upload several files that I have trained back in the days (because I can't have any time for re-train this)

SangbumChoi commented 1 year ago

https://drive.google.com/drive/folders/146ZFPZyFyRQejB8CBYZ_R26NEXEO4EjI?usp=share_link

Hi, I have shared pytorch model, but you should know that these files may need some minor modification in order to do proper inference (such as size modification).

ALISCIFP commented 1 year ago

@SangbumChoi Hi Dr. Choi, Thanks fo sharing for these. Can you specify which file corresponds to which experient configuration in your paper? I mean the correspondence between these pre-trained PyTorch models and experiment configurations. Thanks!

SangbumChoi commented 1 year ago

@ALISCIFP I just uploaded random pth file that I have (I have more random ONNX, TF file also). So I'm not sure which one is the exact configuration of mine... You can re-run my train code.

ALISCIFP commented 1 year ago

@SangbumChoi Thanks so much! That's fine. Can you upload all the ONNX, and TF files, please?

SangbumChoi commented 1 year ago

https://github.com/PINTO0309/PINTO_model_zoo/tree/main/156_MobileHumanPose