SAR ship target detection method based on CNN structure with wavelet and attention mechanism

PLoS One. 2022 Jun 3;17(6):e0265599. doi: 10.1371/journal.pone.0265599. eCollection 2022.

Abstract

Ship target detection in synthetic aperture radar (SAR) images is an important application field. Due to the existence of sea clutter, especially the SAR imaging in huge wave area, SAR images contain a lot of complex noise, which brings great challenges to the effective detection of ship targets in SAR images. Although the deep semantic segmentation network has been widely used in the detection of ship targets in recent years, the global information of the image cannot be fully utilized. To solve this problem, a new convolutional neural network (CNN) method based on wavelet and attention mechanism was proposed in this paper, called the WA-CNN algorithm. The new method uses the U-Net structure to construct the network, which not only effectively reduces the depth of the network structure, but also significantly improves the complexity of the network. The basic network of WA-CNN algorithm consists of encoder and decoder. Dual tree complex wavelet transform (DTCWT) is introduced into the pooling layer of the encoder to smooth the speckle noise in SAR images, which is beneficial to preserve the contour structure and detail information of the target in the feature image. The attention mechanism theory is added into the decoder to obtain the global information of the ship target. Two public SAR image datasets were used to verify the proposed method, and good experimental results were obtained. This shows that the method proposed in this article is effective and feasible.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Neural Networks, Computer
  • Radar*
  • Ships*
  • Wavelet Analysis

Grants and funding

This study was funded by the Natural Science Key Basic Research Plan in Shaanxi Province of China in the form of a grant (2020JZ-57) to SH. This study was also funded by the Natural Science Foundation of China in the form of grants (61379031, 41574008, 61673017 & 61905285) to SH. The funders had no role in study design, data collection, and analysis, decision to publish, or preparation of the manuscript.