Ship target detection in SAR images based on SimAM attention YOLOv8
Yuqiao Xu,Wei Du,Lewu Deng,Yi Zhang,Wanli Wen
DOI: https://doi.org/10.1049/cmu2.12837
IF: 1.345
2024-09-21
IET Communications
Abstract:1. Mitigated the uniform treatment disparity across all regions of images processed by the YOLOv8 model, and validated the effectiveness of our method through experiments. 2. The combination of a simple, parameter‐free attention module (SimAM) attention mechanism with the YOLOv8 visual model for synthetic aperture radar ship target detection effectively suppresses background interference. 3. By allocating three‐dimensional weights in convolutional neural networks through an energy function, the detection performance of the model has been improved. Deep learning has been widely applied in ship detection in synthetic aperture radar (SAR) imagery due to their powerful feature representation capabilities. However, YOLOv8 models treat all regions of the image equally during convolutional feature processing, resulting in less‐than‐ideal outcomes. To address this limitation, this study proposes a simple, parameter‐free attention module (SimAM) attention‐based YOLOv8 algorithm for ship detection in SAR images. The proposed algorithm first passes through a backbone network, which incorporates SimAM attention modules. The SimAM attention mechanism successfully allocates the convolutional neural network's 3D weights effectively using an energy function method, without introducing additional parameters. This mechanism enables the network to automatically emphasize key features in the image, enhancing its ability to represent target areas and suppress background interference. Subsequently, deep features are upsampled and fused with relatively shallow features to extract features at three different scales and achieve target detection, ultimately outputting classification and positional information of the targets. The effectiveness of the model on the SAR‐ship‐dataset is experimentally validated achieving an mAP50 value of 97.72% and an mAP50‐95 value of 68.99%, confirming the superiority of the proposed model.
engineering, electrical & electronic