royzon / rotate-yolov3

Arbitrary oriented object detection implemented with yolov3 (attached with some tricks).

I'll release latest code and training pipeline as I return school at request, cause codes are left there, this repo is just a backup commit.

Rotated-Yolov3

Rotaion object detection implemented with yolov3.

Not good enough yet, reach only Hmean 70 on ICDAR15 dataset.

I'll not keep updating here, but PRs are welcomed. Better detector for rotation object detection will be published in my repo as soon as possible(that's why I deprecated ryolo).

Support

• SEBlock
• CUDA RNMS
• riou loss
• Inception module
• DCNv2
• ORN
• SeparableConv
• Mish/Swish
• GlobalAttention

Notice

Feel free to contact me if you have any question when use this code, cause maybe I don't know either.(too long the last time I make modification on it, and I don't think yolo is a good choice for arbitrary orientation object detection.)
I'll release a stronger detector later.

Some Results

Q&A

Following questions are frequently mentioned. And if you have something unclear, don't doubt and contact me via opening issues.

• Q: How can I obtain icdar_608_care.txt?

  A: icdar_608_care.txt sets the initial anchors generated via kmeans, you need to run kmeans.py refer to my implemention here. You can also check utils/parse_config.py for more details.

• Q: How to train the model on my own dataset?

  A: This ryolo implemention is based on this repo, training and evaluation pipeline are the same as that one do.

• Q: Where is ORN codes?

  A: I'll release the whole codebase as I return school, and this repo may help.

• Q: I cannot reproduce the result you reported(80 mAP for hrsc and 0.7 F1 for IC15).

• A: Refer to my reply here. This is only a backup repo, the overall model is no problem, but direct running does not necessarily guarantee good results, cause it is not the latest version, and some parameters may have problems, you need to adjust some details and parameter settings yourself. I will upload the complete executable code as soon as I return to school in September (if lucky).