Improved YOLOv4 Based on Attention Mechanism for Ship Detection in SAR Images

Ship detection in synthetic aperture radar (SAR) images is an important and challenging work in the field of image processing. Traditional detection algorithms usually rely on handmade features or predefined thresholds, the different performance is obtained with varying degrees of prior knowledge, and it is difficult to take advantage of big data. Recently, deep learning algorithms have found wide applications in ship detection from SAR images. However, due to the complex backgrounds and multiscale ships, it is hard for deep networks to extract representative target features, which limits the ship detection performance to a certain extent. In order to tackle the above problems, we propose an improved YOLOv4 (ImYOLOv4) based on attention mechanism. Firstly, to achieve the best trade-off between detection accuracy and speed, we adopt the off-the-shelf YOLOv4 as our basic framework because of its fast detection speed. Secondly, a thresholding attention module (TAM) is introduced to suppress the adverse effect of complex backgrounds and noises. Besides, we embed channel attention module (CAM) into improved BiFPN as the feature pyramid network (FPN) to better enhance the discrimination of the multiscale target features. Finally, the decoupled head with two parallel branches improves the performance of classification and regression. The proposed method is evaluated on public SAR dataset and the experimental results demonstrate that it has higher efficiency and feasibility than other mainstream methods, yielding the accuracy of 94.16% at intersection over union of 0.5 and 58.19% at intersection over union of 0.75.

Automated summary

Ship detection in SAR images is an important task in image processing. Traditional algorithms rely on handmade features or predefined thresholds, while deep learning algorithms have shown promise. This paper proposes an improved YOLOv4 algorithm based on attention mechanism to enhance ship detection performance. Experimental results demonstrate that the proposed method achieves high accuracy and efficiency on a public SAR dataset.