Yolov5 for Oriented Object Detection

The code for the implementation of “Yolov5 + Circular Smooth Label”.

Results and Models

The results on DOTA_subsize1024_gap200_rate1.0 test-dev set are shown in the table below. (password: yolo)

Model ^(link)	Size ^(pixels)	TTA ^{(multi-scale/ rotate testing)}	OBB mAP^{test ^0.5 DOTAv1.0}	OBB mAP^{test ^0.5 DOTAv1.5}	OBB mAP^{test ^0.5 DOTAv2.0}	Speed ^{CPU b1 (ms)}	Speed ^{2080Ti b1 (ms)}	Speed ^{2080Ti b16 (ms)}	params ^(M)	FLOPs ^{@1024 (B)}
yolov5m [baidu/google]	1024	×	77.30	73.19	58.01	328.2	16.9	11.3	21.6	50.5
yolov5m6	1024	×	-	-	-	-	-	-	-	-
yolov5m7	1024	×	-	-	-	-	-	-	-	-

Table Notes (click to expand / **点我看更多**)

All checkpoints are trained to 300 epochs with COCO pre-trained checkpoints, default settings and hyperparameters.
mAP^{test dota} values are for single-model single-scale on DOTA(1024,1024,200,1.0) dataset.
Reproduce Example:

python val.py --data 'data/dotav15_poly.yaml' --img 1024 --conf 0.01 --iou 0.4 --task 'test' --batch 16 --save-json --name 'dotav15_test_split'
python tools/TestJson2VocClassTxt.py --json_path 'runs/val/dotav15_test_split/best_obb_predictions.json' --save_path 'runs/val/dotav15_test_split/obb_predictions_Txt'
python DOTA_devkit/ResultMerge_multi_process.py --scrpath 'runs/val/dotav15_test_split/obb_predictions_Txt' --dstpath 'runs/val/dotav15_test_split/obb_predictions_Txt_Merged'
zip the poly format results files and submit it to https://captain-whu.github.io/DOTA/evaluation.html

Speed averaged over DOTAv1.5 val_split_subsize1024_gap200 images using a 2080Ti gpu. NMS + pre-process times is included.
Reproduce by python val.py --data 'data/dotav15_poly.yaml' --img 1024 --task speed --batch 1

Updates

[2022/1/7] : Faster and stronger, some bugs fixed, yolov5 base version updated.

Installation

Please refer to install.md for installation and dataset preparation.

Getting Started

This repo is based on yolov5.

And this repo has been rebuilt, Please see GetStart.md for the Oriented Detection latest basic usage.

Acknowledgements

I have used utility functions from other wonderful open-source projects. Espeicially thank the authors of:

More detailed explanation

想要了解相关实现的细节和原理可以看我的知乎文章:

自己改建YOLOv5旋转目标的踩坑记录.

有问题反馈

在使用中有任何问题，欢迎反馈给我，可以用以下联系方式跟我交流

知乎（@略略略）
代码问题提issues,其他问题请知乎上联系

关于作者

  Name  : "胡凯旋"
  describe myself："咸鱼一枚"

zhao521zzj/yolov5_obb