XinzeLee / RotateObjectDetection

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

Pretrained models ? #15

Open QuanNguyen94 opened 2 years ago

QuanNguyen94 commented 2 years ago

Hi, I'm really excited with your work, I want to train with my custom dataset, do you have a strong pretrained model for this rotated version ? Btw, I confuse about the label a little bit, I know the format is "cls_is,cx,cy,w,h,sin(theta),cos(theta)", but what w,h here really stand for in rotated box ? And how you set the rule for theta, where does it start, which direction, does it have bound up, bound down ? Really useful if you have an image to explain all, thank for your great work.