Learning Modulated Loss for Rotated Object Detection

Wen Qian,Xue Yang,Silonge Peng,Junchi Yan,Yue GuO
DOI: https://doi.org/10.1609/aaai.v35i3.16347
2021-01-01
Proceedings of the AAAI Conference on Artificial Intelligence
Abstract:Popular rotated detection methods usually use five parameters (coordinates ofthe central point, width, height, and rotation angle) to describe the rotatedbounding box and l1-loss as the loss function. In this paper, we argue that theaforementioned integration can cause training instability and performancedegeneration, due to the loss discontinuity resulted from the inherentperiodicity of angles and the associated sudden exchange of width and height.This problem is further pronounced given the regression inconsistency amongfive parameters with different measurement units. We refer to the above issuesas rotation sensitivity error (RSE) and propose a modulated rotation loss todismiss the loss discontinuity. Our new loss is combined with theeight-parameter regression to further solve the problem of inconsistentparameter regression. Experiments show the state-of-art performances of ourmethod on the public aerial image benchmark DOTA and UCAS-AOD. Itsgeneralization abilities are also verified on ICDAR2015, HRSC2016, and FDDB.Qualitative improvements can be seen in Fig 1, and the source code will bereleased with the publication of the paper.
What problem does this paper attempt to address?