This link provides a Google Colab notebook for inference of CenterNet HourGlass104 Keypoints 512x512 for object detection and pose key point detection. Is there a similar notebook or tutorial to train object detection and key point detection on custom datasets?
-
You will find everything you need in this documentation: tensorflow-object-detection-api-tutorial.readthedocs.io/en/…– David BaceljCommented Aug 31, 2020 at 19:03
-
I only see posts on inferring keypoint locations from pre-trained models. How to train them on custom data? I don't see a tutorial or guide for that matter.– user3317287Commented Aug 31, 2020 at 19:28
Add a comment
|
1 Answer
I have created a detailed 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.