How to train on custom dataset (no humans)

michalfaber / tensorflow_Realtime_Multi-Person_Pose_Estimation

Multi-Person Pose Estimation project for Tensorflow 2.0 with a small and fast model based on MobilenetV3

Other

215 stars 65 forks source link

Open Cyril9227 opened 3 years ago

Cyril9227 commented 3 years ago

Hi,

I'm very much interested in using your work for my personal dataset. However I have 2 constraints :

Any advice on how to train a model on this custom dataset ? I believe I should remove the paf layers and only keep the heatmap...

Cheers