google-coral / project-bodypix

BodyPix model demo application for Google Coral
Apache License 2.0
275 stars 52 forks source link

Retraining Bodypix Models or Posenet #31

Closed KushalHM closed 2 years ago

KushalHM commented 2 years ago

Description

Hi, I would like to retrain the posenet component of bodypix with some additional keypoint labelled images and also with some augmented images (images rotated at certain degree).

This is to enhance the capability of the posenet component to better detected inverted (fallen down) person or a person spotted at a particular angle relative to the camera view.

Is there any to retrain these models, like training script with pretrained model? If not is there any way to improve the accuracy of these models to better detect fallen down humans?

Click to expand! ### Issue Type Feature Request ### Operating System Ubuntu ### Coral Device Dev Board ### Other Devices _No response_ ### Programming Language C++, Python 3.8 ### Relevant Log Output _No response_
hjonnala commented 2 years ago

Hello @KushalHM Unfortunately, we don't have any scripts or support to train a bodypix model. Please check this link and reach out to tensorflow team for any support. Thanks!

KushalHM commented 2 years ago

Thanks for the quick response. I will reach out to tensorflow team.

google-coral-bot[bot] commented 2 years ago

Are you satisfied with the resolution of your issue? Yes No