'get object rotation to complement the model's output and reproduce the objects with direction and coordinates from Object detection API tf2

Does anybody know if the Object Detection API tf2 provides the rotation details to plot the object in its correct direction?

I've trained a model with the Object Detection API from tensorflow using MASK_RCNN as pre trained model. I'm using bbox and thinking that this might be an issue to get the rotation of my objects and maybe I need to solve this during the annotation step.

From my output I get the objects coordinates, ID, name.. however I also would like to have the object direction. e.g.: {'box': array([0.7090868 , 0.23337238, 0.7342466 , 0.25260702], dtype=float32), 'class_name': 'pump', 'score': 0.9815392}



Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source