LI Nan, YE Xiaodong, WANG Hao, HUANG Xinyu, TAO Shifei. A Ship Detection Method for SAR Images in Complex Scene Based on Improved YOLOv5[J]. JOURNAL OF SIGNAL PROCESSING, 2022, 38(5): 1009-1018. DOI: 10.16798/j.issn.1003-0530.2022.05.013
Citation: LI Nan, YE Xiaodong, WANG Hao, HUANG Xinyu, TAO Shifei. A Ship Detection Method for SAR Images in Complex Scene Based on Improved YOLOv5[J]. JOURNAL OF SIGNAL PROCESSING, 2022, 38(5): 1009-1018. DOI: 10.16798/j.issn.1003-0530.2022.05.013

A Ship Detection Method for SAR Images in Complex Scene Based on Improved YOLOv5

  • The traditional method of synthetic aperture radar image for ship detection is easily affected by complex background, resulting in the number increasing of missed targets. In this paper, the adaptive attention module composed of channel attention and spatial attention is employed for the YOLOv5 network. Most of network is allocated to target feature by using the weighted feature vectors in which the network’s ability of feature learning has been enhanced. Then the loss function of the detection model is optimized where the confidence of prediction boxes has been improved. Also the false negative probability has been reduced for target detecting in the complex background. According to simulations, the recall rate has been improved significantly by the proposed algorithm. The AP (Average Precision) value of ship detection for SAR image in complex scene has reached up to 79.8%, which increased by 26.1% and 17.3% respectively compared with the YOLOv5 algorithm and Faster R-CNN algorithm.
  • loading

Catalog

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return