Adding Keypoint Detection to SSD

[aidansmyth95]

Hi all,

I would like to add a keypoint estimation head to the SSD model (similar to centernet's similar component). However, I cannot figure out how to create the proto necessary to do this.

Please provide some advice or example on what I need to do to create a custom model that is SSD but incorporates this additional head. I plan to train from scratch.

Thanks in advance!

[aidansmyth95]

Additional comment - I want to learn how to create and train custom models like this using TF OD API. So far I have not been able to find any documentation on this topic.

[Petros626]

Hi all,

I would like to add a keypoint estimation head to the SSD model (similar to centernet's similar component). However, I cannot figure out how to create the proto necessary to do this.

Please provide some advice or example on what I need to do to create a custom model that is SSD but incorporates this additional head. I plan to train from scratch.

Thanks in advance!

So I guess you have read this link: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/defining_your_own_model.md
Did you follow all the steps mentioned here?

[Petros626]

Additional comment - I want to learn how to create and train custom models like this using TF OD API. So far I have not been able to find any documentation on this topic.

That's impossible they provide a tutorial, also you find enough tutorials on the web for this task. Just search enough and you will find