Journal of Jilin University (Information Science Edition) ›› 2024, Vol. 42 ›› Issue (6): 1048-1057.

Previous Articles     Next Articles

Optical Remote Sensing Ship Small Target Detection Based on UPCBAM-RYOLO V5

YANG Xiaotian1,2, YU Xin1,2, LIU Ming3, WANG Liang1,2, TAN Jinlin1,2, WU Yi1,2    

  1. 1. China Academy of Space Technology(Xi’an), Xian 710100, China; 2. General Department of the System, Shaanxi Aerospace Technology Application Research Institute Company Limited, Xian 710100, China; 3. School of Mathematics and Statistics, Changchun University of Technology, Changchun 130012, China
  • Received:2023-11-07 Online:2024-12-23 Published:2024-12-23

Abstract: In order to solve the problems of large proportion of small targets in optical remote sensing data, the aspect ratio is large, and multiple targets are closely arranged and difficult to detect,we present an optical remote sensing small ship target detection algorithm based on the YOLO V5(You Only Look Once V5), UPCBAM- RYOLO V5 (Upsampling Convolutional Attention Block Module-RYOLO V5) algorithm. An up-sampling attention mechanism module is designed to enhance the feature extraction ability of small size targets. The rotation angle loss is introduced into the frame regression formula to improve the algorithm’s perception ability of the ships appearance and direction. Based on the experiment of small ship target datasets composed of GF1 and GF2, the results show that the UPCBAM-RYOLO V5 algorithm model improves the positioning accuracy and classification accuracy of small ship target detection. The P value, R value, and MAP(Mean Average Precision) value reach 93%, 94%, and 95% respectively, which are 3%, 7%, and 7% higher than the original YOLO V5 model. In the upsampled attention-mechanism module added location ablation experiment to the network, the results show that compared with the addition of UPCBAM in Backbone and Prediction, the addition of UPCBAM in Neck has the greatest influence on the detection of small target ships in remote sensing images. The performance is the best, with P value, R value, and MAP value increased by 5%, 4%, and 2%, respectively. The UPCBAM-RYOLO V5 model is proved to have a certain research significance in optical remote sensing ship small target detection.

Key words: optical remote sensing data, small ship targets, upsampling convolutional attention block module-R you only look once V5(UPCBAM-RYOLOV5) algrithom, upsampling convolutional attention block module, rotation angle

CLC Number: 

  • TP319.4