BFEA: A SAR Ship Detection Model Based on Attention Mechanism and Multiscale Feature Fusion
Liming Zhou, Z. Wan, Shuai Zhao, Hongyu Han, Yang Liu
Abstract
With the advancements in deep learning and Synthetic Aperture Radar (SAR) technology, an increasing number of individuals are utilizing deep learning techniques to detect ships in SAR images. However, the efficiency of SAR ship detection is affected by complex background interference and various ship sizes. Addressing these challenges, this paper proposes a balanced feature-enhanced attention model (BFEA). First, we introduce a novel Attention Feature Fusion Network (WEF-Net) tailored for SAR multi-scale ship detection. WEF-Net effectively balances the information across different backbone layers and harmonizes semantic information from various levels of the feature pyramid through aggregation and averaging. Next, we embed the receiving field extension module (MRFFM) in WEF-Net to learn the context information and generate the global characteristics of the receiving field balance. In addition, it can extract features from multiple scales to enhance the detection capability of the model for ships of different scales. At the same time, acknowledging the impact of surrounding complex background interference on the detector, we redesigned the ELAN module by combining convolution and attention, This enhancement enables the model to better attend to target position information during feature fusion, suppress the surrounding complex background interference, and highlight the ship's feature information. Finally, owing to the prevalence of small targets in SAR images, we employ an optimized loss function to bolster the model's performance in detecting small targets. This approach accelerates training convergence, reduces instances of missed detection on small targets, and enhances overall detection performance across multiple scales. Experimental results demonstrate that our model achieves detection accuracies of 98%, 93.1%, and 76.9% on the SSDD, HRSID, and LS-SSDD-v1.0 datasets, respectively, effectively discerning ship targets amidst complex backgrounds in SAR images.