XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.
113 stars 21 forks source link

Theta calculation and label normalization #18

Open Jcastanyo opened 2 years ago

Jcastanyo commented 2 years ago

Hello, my name is Julio,

First of all, I wanted to thank you for you hard work on this project.

I am working on a project where rotated-yolov5 should be the optimal solution. I am struggling with theta calculation and label normalization. I wonder if you could explain me how do you calculate theta angle from rectangular bounding boxes and, also, how do you normalize w,h, if you do it.

Thanks in advance.