seeker1943 / rotation-yolov5

rotation detection based on yolov5

the rotation detection

Requirement

yolov5's Requirement
shapely==1.7.1
opencv==4.2.0.34

inference

you can download the weights(password is 4ud5) for ship detection by my dataset(not DOTA) to test the demo.

$ python detect.py

train

what format my model need

Not much different from yolo dataset,just add an angle and we define the box attribute w is always longer than h!

So wo define the box label is (cls, c_x, c_y, Longest side，short side, angle)

Attention!we define angle is a classify question,so we define 180 classes for angle.

For Example: Range for angle is [-90，90), so wo should __add 90 __in angle while make your dataset label and then your label's Range should be [0,179)

$ python train.py

details

If you have any question,welcome discuss with me by This or email to [e-mail]