Learning Modulated Loss for Rotated Object Detection
Wen Qian, Xue Yang, Silong Peng, Junchi Yan, Yue Guo
Abstract
Popular rotated detection methods usually use five parameters (coordinates of the central point, width, height, and rotation angle) to describe the rotated bounding box and 1 loss as the loss function. In this paper, we argue that the aforementioned integration can cause training instability and performance degeneration, due to the loss discontinuity resulted from the inherent periodicity of angles and the associated sudden exchange of width and height. This problem is further pronounced given the regression inconsistency among five parameters with different measurement units. We refer to the above issues as rotation sensitivity error (RSE) and propose a modulated rotation loss to dismiss the loss discontinuity. Our new loss is combined with the eight-parameter regression to further solve the problem of inconsistent parameter regression. Experiments show the state-of-art performances of our method on the public aerial image benchmark DOTA and UCAS-AOD. Its generalization abilities are also verified on ICDAR2015, HRSC2016, and FDDB. Qualitative improvements can be seen in Fig. 1 , and the source code will be released with the publication of the paper.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers25
- Oriented R-CNN for Object DetectionXingxing Xie, Gong Cheng, Jiabao Wang, Xiwen Yao et al.ICCV 2021 · 1,070 citations
- Learning High-Precision Bounding Box for Rotated Object Detection via Kullback-Leibler DivergenceXue Yang, Xiaojiang Yang, Jirui Yang, Qi Ming et al.NeurIPS 2021 · 603 citations
- Rethinking Rotated Object Detection with Gaussian Wasserstein Distance LossXue Yang, Junchi Yan, Qi Ming, Wentao Wang et al.ICML 2021 · 572 citations
- Large Selective Kernel Network for Remote Sensing Object DetectionYuxuan Li, Qibin Hou, Zhaohui Zheng, Ming-Ming Cheng et al.ICCV 2023 · 535 citations
- Oriented RepPoints for Aerial Object DetectionWentong Li, Yijie Chen, Kaixuan Hu, Jianke ZhuCVPR 2022 · 487 citations
Builds on4
- FCOS: Fully Convolutional One-Stage Object DetectionZhi Tian, Chunhua Shen, Hao Chen, Tong HeICCV 2019 · 6,042 citations
- R3Det: Refined Single-Stage Detector with Feature Refinement for Rotating ObjectXue Yang, Junchi Yan, Ziming Feng, Tao HeAAAI 2021 · 1,109 citations
- RepPoints: Point Set Representation for Object DetectionZe Yang, Shaohui Liu, Han Hu, Liwei Wang et al.ICCV 2019 · 1,056 citations
- SCRDet: Towards More Robust Detection for Small, Cluttered and Rotated ObjectsXue Yang, Jirui Yang, Junchi Yan, Yue Zhang et al.ICCV 2019 · 865 citations
Related papers
- Dense Label Encoding for Boundary Discontinuity Free Rotation DetectionXue Yang, Liping Hou, Yue Zhou, Wentao Wang et al.CVPR 2021
- OSKDet: Orientation-sensitive Keypoint Localization for Rotated Object DetectionDongchen Lu, Dongmei Li, Yali Li, Shengjin WangCVPR 2022 · 25 citations
- ReDet: A Rotation-Equivariant Detector for Aerial Object DetectionJiaming Han, Jian Ding, Nan Xue, Gui-Song XiaCVPR 2021
- Dynamic Anchor Learning for Arbitrary-Oriented Object DetectionQi Ming, Zhiqiang Zhou, Lingjuan Miao, Hongwei Zhang et al.AAAI 2021 · 332 citations
- PointOBB: Learning Oriented Object Detection via Single Point SupervisionJunwei Luo, Xue Yang, Yi Yu, Qingyun Li et al.CVPR 2024 · 41 citations
