Research on Improved YOLO11 for Detecting Small Targets in Sonar Images Based on Data Enhancement
Xiaochuan Wang, Zhi‐Qiang Zhang, Xiaodong Shang
Abstract
Existing sonar target detection methods suffer from low efficiency and accuracy due to sparse target features and significant noise interference in sonar images. To address this, we introduce SFE-YOLO, an improved model based on YOLOv11. We replace the original detection head with an FSAFFHead module that enables adaptive spatial feature fusion. An EEA module is designed to direct the model’s attention to the intrinsic contour information of targets. We also enhance SC_Conv convolution and integrate it into C3K2 to improve detection stability and reduce information redundancy. Additionally, Focaler-IOU is introduced to boost the accuracy of multi-category target bounding box regression. Lastly, we employ a hybrid training strategy that combines pre-training with ADA-StyleGAN3-generated data and transfer learning with real data to alleviate the problem of insufficient training samples. The experiments show that, compared to the baseline YOLOv11n, the improved model’s precision and recall increase to 92% and 90.3%, respectively, and mAP50 rises by 12.7 percentage points, highlighting the effectiveness of the SFE-YOLO network and its transfer learning strategy in tackling the challenges of sparse small target features and strong noise interference in sonar images.