S$^{2}$O-Det: A Semisupervised Oriented Object Detection Network for Remote Sensing Images

Ronghao Fu,Shuang Yan,Chengcheng Chen,Xianchang Wang,Ali Asgher Heidari,Jing Li,Huiling Chen
DOI: https://doi.org/10.1109/tii.2024.3403260
IF: 12.3
2024-01-01
IEEE Transactions on Industrial Informatics
Abstract:Semisupervised object detection (SSOD) has garnered significant interest for its capability to enhance the detection performance by leveraging large amounts of unlabeled data. However, current SSOD methods primarily focus on detecting horizontal objects, with little research devoted to the detection of arbitrary-oriented objects in remote sensing images. Drawing inspiration from this limitation, this article proposes a semisupervised oriented object detection framework (S $^{2}$ O-Det) to reduce annotation costs while improving detection performance in a semisupervised manner. Initially, the proposed task-consistent learning aims to alleviate the inconsistencies between classification and localization, which provides consistent confidence for the pseudolabels. Subsequently, the introduced coarse-to-fine sample mining employs dense prediction for pseudolabel assignment, adopting a divide-and-conquer approach to independently identify consistent and reliable labels for both classification and localization tasks. Finally, a probabilistic distillation loss ensures the harmonization of the probability distributions across the teacher and student feature domains, thereby reciprocally enhancing the learning competencies. Experimental results on the DOTA-v1.0 and DOTA-v1.5 datasets demonstrate that S $^{2}$ O-Det achieves promising performance across different labeling ratios.
What problem does this paper attempt to address?