21 June 2023 MKIoU loss: toward accurate oriented object detection in aerial images
Xinyi Yu, Jiangping Lu, Mi Lin, Libo Zhou, Linlin Ou
Author Affiliations +
Abstract

Oriented bounding box regression is crucial for oriented object detection. However, regression-based methods often suffer from boundary problems and the inconsistency between loss and evaluation metrics. A modulated Kalman intersection over union (IoU) loss of approximate SkewIoU is proposed, named MKIoU. To avoid boundary problems, we convert the oriented bounding box to Gaussian distribution then use the Kalman filter to approximate the intersection area. However, there exists significant difference between the calculated and actual intersection areas. Thus, we propose a modulation factor to adjust the sensitivity of angle deviation and width-height offset to loss variation, making the loss more consistent with the evaluation metric. Furthermore, the Gaussian modeling method avoids the boundary problem but causes the angle confusion of square objects simultaneously. Thus, the Gaussian angle loss (GA loss) is presented to solve this problem by adding a corrected loss for square targets. The proposed GA loss can be easily extended to other Gaussian-based methods. Experiments on three publicly available aerial image datasets, DOTA, UCAS-AOD, and HRSC2016, show the effectiveness of the proposed method.

© 2023 SPIE and IS&T
Xinyi Yu, Jiangping Lu, Mi Lin, Libo Zhou, and Linlin Ou "MKIoU loss: toward accurate oriented object detection in aerial images," Journal of Electronic Imaging 32(3), 033030 (21 June 2023). https://doi.org/10.1117/1.JEI.32.3.033030
Received: 23 August 2022; Accepted: 26 May 2023; Published: 21 June 2023
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Education and training

Modeling

Modulation

Ablation

Transformers

Performance modeling

Back to Top