models icon indicating copy to clipboard operation
models copied to clipboard

Add rotated bounding boxes and anchors for models

Open Petros626 opened this issue 2 years ago • 3 comments

Hello dear TensorFlow-team,

I wanted to ask if there are any plans to add a long awaited feature to the Object Detection API in the near future. It is about rotated bounding boxes for training and execution of models.

Nvidia has already taken the step here and the code for implementing this feature is certainly very helpful. (https://github.com/NVIDIA/retinanet-examples/blob/main/odtk/box.py)

On behalf of the community, we would all certainly be very happy to receive this feature.

Greetings,

Petros

Petros626 avatar Jan 16 '23 11:01 Petros626

@fyangf,

Could you please look into this issue in your free time.

Thanks.

laxmareddyp avatar Jan 17 '23 22:01 laxmareddyp

Object Detection API is now maintained by the community so you are welcome to contribute to this feature. If you are talking about the models under official, then we do not have a plan to support rotated box in the short term. However, if you really care about rotated box, you may take a look at centernet: https://github.com/tensorflow/models/tree/master/official/projects/centernet, which is anchor free and can naturally handle rotated boxes (some code change may be needed).

fyangf avatar Feb 10 '23 18:02 fyangf

Object Detection API is now maintained by the community so you are welcome to contribute to this feature. If you are talking about the models under official, then we do not have a plan to support rotated box in the short term. However, if you really care about rotated box, you may take a look at centernet: https://github.com/tensorflow/models/tree/master/official/projects/centernet, which is anchor free and can naturally handle rotated boxes (some code change may be needed).

Hey, thanks for the reply.

In principle, I think it would make sense for Tensorflow to equip all models with this feature, but this requires renewed training (Coco dataset) of all models with the rotated boxes, which could be made available to the community. I lack the hardware and time to train for all common models from scratch.

So this feature will probably not appear, right?

Petros626 avatar Feb 10 '23 18:02 Petros626