Skip to content

BossZard/rotation-yolov5

Repository files navigation

the rotation detection

Requirement

torch==1.6
shapely==1.7.1
opencv==4.2.0.34

inference

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

$ python detect.py

image

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: image 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) image

modify yaml

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

update

2021.1.4---correct some BUG for training

details

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

About

rotation detection based on yolov5

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published