An Improved Lightweight RetinaNet for Ship Detection in SAR Images

Abstract

The rapid development of remote sensing technology has led to a sharp increase in the amount of synthetic aperture radar (SAR) measurements, which put forward higher requirements for remote sensing image processing. As an important application of SAR, fast and accurate ship detection has always been a research hotspot. In this article, an improved lightweight RetinaNet for ship detection in SAR images is proposed. Compared with the standard RetinaNet, the shallow convolutional layers of the backbone are replaced by ghost modules and the number of the deep convolutional layers is reduced. The spatial and channel attention modules are embedded into the model to enhance detectability. K-means clustering algorithm is applied to adjust the initial aspect ratios of the model. The effectiveness and robustness of the proposed method is demonstrated by numerical experiments with SSDD dataset, Gaofen-3 mini dataset, and a large-scale SAR image of Hisea-1 satellite, it is shown that the proposed method can significantly reduce the floating-point operations and the number of parameters without decreasing the detection accuracy and recall ratio. Moreover, the experimental results also show the proposed model's robustness and the ability to detect ship targets in small datasets.

Keywords

• Lightweight design
• Retinanet
• synthetic aperture radar (SAR)
• ship detection