PAG-YOLO: A Portable Attention-Guided YOLO Network for Small Ship Detection

The YOLO network has been extensively employed in the field of ship detection in optical images. However, the YOLO model rarely considers the global and local relationships in the input image, which limits the final target prediction performance to a certain extent, especially for small ship targets. To address this problem, we propose a novel small ship detection method, which improves the detection accuracy compared with the YOLO-based network architecture and does not increase the amount of computation significantly. Specifically, attention mechanisms in spatial and channel dimensions are proposed to adaptively assign the importance of features in different scales. Moreover, in order to improve the training efficiency and detection accuracy, a new loss function is employed to constrain the detection step, which enables the detector to learn the shape of the ship target more efficiently. The experimental results on a public and high-quality ship dataset indicate that our method realizes state-of-the-art performance in comparison with several widely used advanced approaches.

[1]  Fang Xu,et al.  Ship Detection in Optical Remote Sensing Images Based on Saliency and a Rotation-Invariant Descriptor , 2018, Remote. Sens..

[2]  Dongping Ming,et al.  Aircraft Detection in High Spatial Resolution Remote Sensing Images Combining Multi-Angle Features Driven and Majority Voting CNN , 2021, Remote. Sens..

[3]  Hao Zhou,et al.  Joint Ship Detection Based on Time-Frequency Domain and CFAR Methods with HF Radar , 2021, Remote. Sens..

[4]  Lukasz Kaiser,et al.  Attention is All you Need , 2017, NIPS.

[5]  Haibo Luo,et al.  Adaptive feature fusion with attention mechanism for multi-scale target detection , 2020, Neural Computing and Applications.

[6]  Jie Ma,et al.  Exploit Direction Information for Remote Ship Detection , 2021, Remote. Sens..

[7]  Antonio-Javier Gallego,et al.  Automatic Ship Classification from Optical Aerial Images with Convolutional Neural Networks , 2018, Remote. Sens..

[8]  Xuan Wang,et al.  Ship Detection in Optical Remote Sensing Images Based on Wavelet Transform and Multi-Level False Alarm Identification , 2017, Remote. Sens..

[9]  Hong-Yuan Mark Liao,et al.  YOLOv4: Optimal Speed and Accuracy of Object Detection , 2020, ArXiv.

[10]  Dawei Wang,et al.  Remote sensing image fine-processing method based on the adaptive hyper-Laplacian prior , 2021 .

[11]  Wei Zhang,et al.  Salient Ship Detection via Background Prior and Foreground Constraint in Remote Sensing Images , 2020, Remote. Sens..

[12]  Zongxu Pan,et al.  CPS-Det: An Anchor-Free Based Rotation Detector for Ship Detection , 2021, Remote. Sens..

[13]  Jianlin Wang,et al.  DC-SPP-YOLO: Dense Connection and Spatial Pyramid Pooling Based YOLO for Object Detection , 2019, Inf. Sci..

[14]  Wei Liu,et al.  SSD: Single Shot MultiBox Detector , 2015, ECCV.