arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.09670cs.CVcs.AI
收藏

用于边界不连续性自由旋转检测的稠密标号编码

Dense Label Encoding for Boundary Discontinuity Free Rotation Detection

Shanghai Jiao Tong University(上海交通大学);University of Chinese Academy of Sciences(中国科学院大学)

Xue Yang, Liping Hou, Yue Zhou, Wentao Wang, Junchi Yan

摘要:

旋转检测作为一个基本的积木在许多视觉应用中,涉及航空图像,场景文本,人脸等不同的主导回归为基础的方法进行方向估计,本文探讨了一个相对较少研究的方法分类的基础上。希望是固有地驳回边界不连续性问题所遇到的回归为基础的检测器。我们提出了新的技术来推动其在两个方面的前沿:i)新的编码机制:设计两个用于角度分类的密集编码标签(DCL),以取代现有基于分类的检测器中的稀疏编码标签(SCL),导致训练速度增加三倍,如在基准测试中经验观察到的,进一步显着提高检测精度; ii)损失重新加权:我们提出了角度距离和长宽比敏感加权(ADARSW),通过使基于DCL的检测器对角度距离和物体的长宽比敏感,提高了检测精度,特别是对于方形物体。在大规模公开的航空影像数据集DOTA、UCAS-AOD、HRSC 2016以及场景文本数据集ICDAR 2015和MLT上进行的大量实验和可视化分析表明了该方法的有效性。源代码可在https://github.com/Thinklab-SJTU/DCL_RetinaNet_Tensorflow上获得,并且还集成在我们的开源旋转检测基准中:https://github.com/yangxue0827/RotationDetection。

英文摘要:

Rotation detection serves as a fundamental building block in many visual applications involving aerial image, scene text, and face etc. Differing from the dominant regression-based approaches for orientation estimation, this paper explores a relatively less-studied methodology based on classification. The hope is to inherently dismiss the boundary discontinuity issue as encountered by the regression-based detectors. We propose new techniques to push its frontier in two aspects: i) new encoding mechanism: the design of two Densely Coded Labels (DCL) for angle classification, to replace the Sparsely Coded Label (SCL) in existing classification-based detectors, leading to three times training speed increase as empirically observed across benchmarks, further with notable improvement in detection accuracy; ii) loss re-weighting: we propose Angle Distance and Aspect Ratio Sensitive Weighting (ADARSW), which improves the detection accuracy especially for square-like objects, by making DCL-based detectors sensitive to angular distance and object's aspect ratio. Extensive experiments and visual analysis on large-scale public datasets for aerial images i.e. DOTA, UCAS-AOD, HRSC2016, as well as scene text dataset ICDAR2015 and MLT, show the effectiveness of our approach. The source code is available at https://github.com/Thinklab-SJTU/DCL_RetinaNet_Tensorflow and is also integrated in our open source rotation detection benchmark: https://github.com/yangxue0827/RotationDetection.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑