the rotation detection
torch==1.6
shapely==1.7.1
opencv==4.2.0.34
you can download the weights(password is 4ud5) for ship detection by my dataset(not DOTA) to test the demo.
$ python detect.py
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)
models/yolov5m.yaml: set nc to your dataset class num; data/wheat0.yaml:set nc to your dataset class num, and set names to your dataset class name;
$ python train.py
2021.1.4---correct some BUG for training
If you have any question,welcome discuss with me by This or email to [email protected]