Yan Zexin, Fan Jie, Li Zhongbo, Xie Yongqiang
Institute of System Engineering, Academy of Military Sciences, Beijing 100141, China.
Sensors (Basel). 2024 Aug 18;24(16):5342. doi: 10.3390/s24165342.
In optical remote sensing image object detection, discontinuous boundaries often limit detection accuracy, particularly at high Intersection over Union (IoU) thresholds. This paper addresses this issue by proposing the Spatial Adaptive Angle-Aware (SA3) Network. The SA3 Network employs a hierarchical refinement approach, consisting of coarse regression, fine regression, and precise tuning, to optimize the angle parameters of rotated bounding boxes. It adapts to specific task scenarios using either class-aware or class-agnostic strategies. Experimental results demonstrate its effectiveness in significantly improving detection accuracy at high IoU thresholds. Additionally, we introduce a Gaussian transform-based IoU factor during angle regression loss calculation, leading to the development of Edge-aware Skewed Bounding Box Loss (EAS Loss). The EAS loss enhances the loss gradient at the final stage of angle regression for bounding boxes, addressing the challenge of further learning when the predicted box angle closely aligns with the real target box angle. This results in increased training efficiency and better alignment between training and evaluation metrics. Experimental results show that the proposed method substantially enhances the detection accuracy of ReDet and ReBiDet models. The SA3 Network and EAS loss not only elevate the mAP of the ReBiDet model on DOTA-v1.5 to 78.85% but also effectively improve the model's mAP under high IoU threshold conditions.
在光学遥感图像目标检测中,不连续的边界常常限制检测精度,尤其是在高交并比(IoU)阈值的情况下。本文通过提出空间自适应角度感知(SA3)网络来解决这一问题。SA3网络采用一种分层细化方法,包括粗回归、细回归和精确调整,以优化旋转边界框的角度参数。它使用类别感知或类别不可知策略来适应特定的任务场景。实验结果表明,它在高IoU阈值下能显著提高检测精度。此外,我们在角度回归损失计算过程中引入基于高斯变换的IoU因子,从而开发出边缘感知倾斜边界框损失(EAS损失)。EAS损失增强了边界框角度回归最后阶段的损失梯度,解决了预测框角度与真实目标框角度紧密对齐时进一步学习的挑战。这导致训练效率提高,训练和评估指标之间的对齐性更好。实验结果表明,所提出的方法显著提高了ReDet和ReBiDet模型的检测精度。SA3网络和EAS损失不仅将ReBiDet模型在DOTA-v1.5上的平均精度均值(mAP)提高到78.85%,而且在高IoU阈值条件下有效地提高了模型的mAP。