Infrared Ship Segmentation Based on Weakly-Supervised and Semi-Supervised Learning

Abstract

Existing fully-supervised semantic segmentation methods have achieved good performance. However, they all rely on high-quality pixel-level labels. To minimize the annotation costs, weakly-supervised methods or semi-supervised methods are proposed. When such methods are applied to the infrared ship image segmentation, inaccurate object localization occurs, leading to poor segmentation results. In this paper, we propose an infrared ship segmentation (ISS) method based on weakly-supervised and semi-supervised learning, aiming to improve the performance of ISS by combining the advantages of two learning methods. It uses only image-level labels and a minimal number of pixel-level labels to segment different classes of infrared ships. Our proposed method includes three steps. First, we designed a dual-branch localization network based on ResNet50 to generate ship localization maps. Second, we trained a saliency network with minimal pixel-level labels and many localization maps to obtain ship saliency maps. Then, we optimized the saliency maps with conditional random fields and combined them with image-level labels to generate pixel-level pseudo-labels. Finally, we trained the segmentation network with these pixel-level pseudo-labels to obtain the final segmentation results. Experimental results on the infrared ship dataset collected on real sites indicate that the proposed method achieves 71.18% mean intersection over union, which is at most 56.72% and 8.75% higher than the state-of-the-art weakly-supervised and semi-supervised methods, respectively.

Keywords

• Infrared ship images
• object segmentation
• weakly-supervised learning
• semi-supervised learning
• pixel-level pseudo-labels