Dense Ship Detection in SAR Images via Comprehensive Confidence Score-Based Label Assignment

• Published in: IEEE journal of selected topics in applied earth observations and remote sensing 2024, Vol.17, p.17175-17186
• Main Authors: Zhang, Yu, Wang, Xueqian, Li, Gang, He, You, Song, Zhaohui, Song, Huina
• Format: Article
• Language: English
• Subjects: Artificial neural networks; Boxes; Centrality; convolutional neural network (CNN); dense ship detection; Detectors; Feature extraction; Head; Heating systems; Image quality; Labels; Localization; Marine vehicles; Neural networks; Radar detection; Radar imaging; Radar polarimetry; SAR (radar); Ships; Synthetic aperture radar; synthetic aperture radar (SAR); Target detection; Training
• ISSN: 1939-1404; 2151-1535
• DOI: 10.1109/JSTARS.2024.3419770

Convolutional neural network (CNN) based ship detection in synthetic aperture radar (SAR) images has achieved impressive attention in recent years. In practice, bounding boxes (B-Boxes) of dense inshore ships in training data have large overlapping areas, which lead to indistinct labels and increase the difficulty to train CNN-based detectors with high performance. To address this issue, a comprehensive confidence score-based detector (CCSDet) is proposed in this article. In CCSDet, we refine the ground-truth B-Boxes (labels) in the overlapping regions of dense ships based on the comprehensive confidence score, which considers both the centrality scores of the labels and the prediction quality of the model in the training stage. The refinement of labels reduces the ambiguity of overlapping ship regions in the training process. Besides, a pixel sample selection strategy is introduced to encourage our detection model to focus on both the total dense target areas and the center of dense ships when the localization loss is calculated. Extensive experiments conducted on the public SAR ship datasets show that our method outperforms the existing methods in the case of densely docked ship detection in SAR images.