search

michalfaber / keras_Realtime_Multi-Person_Pose_Estimation

Keras version of Realtime Multi-Person Pose Estimation project
Other
781 stars 371 forks source link

How to change the model for more keypoints? #143

Open lengyuner opened 3 years ago

lengyuner commented 3 years ago

I want to use this model for images of 20+ key points. What should I do？

michalfaber commented 3 years ago

Hi @lengyuner I would start by changing the dataset generator. It should return heatmaps and pafs adjusted to your number of channels (20) You also need to change the number of output channels in the model. The estimation part is the most complex, so you would have to make a lot of small adjustments here and there.

I will write a tutorial on this topic on my blog www.worldestimators.com

lengyuner commented 3 years ago

Thx~ I have changed the model successfully for my data and I may arrange the project pull it in my github.