search

see-- / keras-centernet

A Keras implementation of CenterNet with pre-trained model (unofficial)
MIT License
337 stars 84 forks source link

how to detect a rotational rectangle? #12

Closed helloxueen closed 4 years ago

helloxueen commented 5 years ago

how to detect a rotational rectangle using keypoint detection?

LinyeLi60 commented 5 years ago

how to detect a rotational rectangle using keypoint detection?

I think you can regress the vector from center to four corners of a rectangle.

see-- commented 4 years ago

You could add an additional head to regress the angle (like the one to regress the offsets). It's interesting but not part of CenterNet, which only supports coordinate aligned boxes.