Level-wise Dynamic Label Assignment for Oriented Object Detection.

Yunhui Zhu,Buliao Huang,Xiren Zhou,Huanhuan Chen
DOI: https://doi.org/10.1109/ssci51031.2022.10022276
2022-01-01
Abstract:Oriented object detection has attracted extensive attention due to its wide range of applications. Most oriented detectors predefine multi-scale anchors on feature pyramids to cover objects of various sizes, and assign labels for all anchors to train the detectors. Recent advanced label assignment strategies take the quality of anchors' predictions as the indicator to assign positive/negative samples across all pyramid levels. However, since the model is not sufficiently optimized in early training stages, the predictions might be unreliable, which might make the positive samples arbitrarily selected from various pyramid levels, further reducing the detection performance. To mitigate the interference of unreliable predictions, this paper proposes a Level-wise Dynamic Label Assignment (LDLA) algorithm, which decouples the selection of pyramid levels from unreliable predictions and selects positive/negative samples level by level. Concretely, LDLA first estimates the number of positive samples on each pyramid level based on the statistics of anchors instead of the predictions. Afterwards, LDLA defines score functions to measure the quality of anchors, and selects anchors with higher scores from each level as positive samples. Extensive experiments conducted on DOTA and HRSC2016 datasets demonstrate the effectiveness of the proposed LDLA.
What problem does this paper attempt to address?