How to perform keypoint regression on a custom dataset in Tensorflow?

Question

Is it possible to use CenterNet with MobilenetV2 backbone (TF Lite compatible) to perform keypoints detection on a custom dataset? Is there a tutorial somewhere?

Answer 1

I have something for training keypoint detection model for custom dataset on Centernet model with hourglass backbone.

This github repo Custom Keypoint Detection for dataset preparation, model training and inference on Centernet-hourglass104 keypoint detection model based on Tensorflow Object detection API with examples.

This could help you in training your keypoint detection model on custom dataset.

Any issues related to the project can be raised in the github itself and doubts can be cleared here.