Shape-Dependent Dynamic Label Assignment for Oriented Remote Sensing Object Detection

Oriented remote sensing object detection (ORSOD) has gained increasing significance in both military and civilian applications due to the necessity of accurately identifying objects with varying shapes and orientations in remote sensing data. Traditional ORSOD methods often employ fixed label assignment strategies to differentiate between positive and negative samples. However, most of them frequently overlook the impact of object shape on sample quality, leading to an imbalanced distribution of positive samples and exacerbating the inconsistency between classification and regression tasks, thereby limiting detection performance. To address these challenges, we propose a novel Shape-Dependent Assignment (SDA) method that dynamically differentiates positive and negative samples based on object shape. It introduces a new metric for evaluating sample box quality by considering angular differences relative to ground truth (GT) boxes and adjusts the sample scoring threshold according to the aspect ratio of each GT box. In addition, we present a DIoU-Adaptive Weighting (DAW) module that enhances the interaction between classification and regression tasks by leveraging the Distance-IoU metric. This approach not only balances the quantity of samples but also improves their quality, enabling more effective training schemes for samples of varying qualities. We validate our proposed methods through extensive experiments on three challenging ORSOD datasets: DOTA-1.0, HRSC2016, and UCAS-AOD. The results demonstrate that our approach achieves significant improvements, especially for objects with large aspect ratios.