Small Ship Detection of SAR Images Based on Optimized Feature Pyramid and Sample Augmentation

Abstract

SAR images have become the latest high-resolution imaging equipment, which can monitor the Earth 24 hours a day. More and more deep learning technologies are applied to ship target detection, however, in complex environments, due to the small target of the ship, problems such as false detection and miss detection often occur. For this reason, SSPNet is proposed with several small-target-augmentation strategies to complete the detection of small ships on the sea. This network is an improvement of FPN. The model uses Context Attention Module (CAM), Scale Enhancement Module (SEM), and Scale Selection Module (SSM). CAM introduces the attention heat map, SEM uses the residual module to make the network pay more attention to specific scale targets, SSM introduces deep semantic features into shallow features. Weighted Negative Sampling (WNS) strategy is proposed to enable the network to select more representative samples. These modules make the network more suitable for small target detection. The results on SSDD dataset show that the model is superior to the existing object detection network, and the AP50 reaches 91.57%.