Daniil-Osokin / lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.
Apache License 2.0
653 stars 137 forks source link

Training problems of pre-training models(human-pose-estimation-3d.pth) #80

Closed congcheng-qu closed 2 years ago

congcheng-qu commented 2 years ago

Hi~ I would like to ask what is the training for this pre-training model. Can you provide train.py and data set for training? Thank you very much

Daniil-Osokin commented 2 years ago

Hi! We have used CMU Panoptic dataset to train the model (please see #75). We did not added training code, since it was not ready for the release. This functionality is open for contribution.

congcheng-qu commented 2 years ago

OK , Thank you very much!