A High-Precision Detection Model of Small Objects in Maritime UAV Perspective Based on Improved YOLOv5

• Published in: Journal of marine science and engineering Vol. 11; no. 9; p. 1680
• Main Authors: Yang, Zhilin, Yin, Yong, Jing, Qianfeng, Shao, Zeyuan
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.09.2023
• Subjects: Accuracy; Aggregation; Algorithms; Analysis; Artificial satellites in remote sensing; attention mechanism; Autonomous underwater vehicles; Convolution; Detection; Drone aircraft; Frames per second; Latency; Modules; Neural networks; object detection; Object recognition; Radar systems; Researchers; Sensors; shipborne UAV scenarios; space pyramid pool; Unmanned aerial vehicles; YOLOv5
• ISSN: 2077-1312; 2077-1312
• DOI: 10.3390/jmse11091680

Object detection by shipborne unmanned aerial vehicles (UAVs) equipped with electro-optical (EO) sensors plays an important role in maritime rescue and ocean monitoring. However, high-precision and low-latency maritime environment small-object-detection algorithms remain a major challenge. To address this problem, this paper proposes the YOLO-BEV (“you only look once”–“bird’s-eye view”) model. First, we constructed a bidirectional feature fusion module—that is, PAN+ (Path Aggregation Network+)—adding an extremely-small-object-prediction head to deal with the large-scale variance of targets at different heights. Second, we propose a C2fSESA (Squeeze-and-Excitation Spatial Attention Based on C2f) module based on the attention mechanism to obtain richer feature information by aggregating features of different depth layers. Finally, we describe a lightweight spatial pyramid pooling structure called RGSPP (Random and Group Convolution Spatial Pyramid Pooling), which uses group convolution and random channel rearrangement to reduce the model’s computational overhead and improve its generalization ability. The article compares the YOLO-BEV model with other object-detection algorithms on the publicly available MOBDrone dataset. The research results show that the mAP0.5 value of YOLO-BEV reached 97.1%, which is 4.3% higher than that of YOLOv5, and the average precision for small objects increased by 22.2%. Additionally, the YOLO-BEV model maintained a detection speed of 48 frames per second (FPS). Consequently, the proposed method effectively balances the accuracy and efficiency of object-detection in shipborne UAV scenarios, outperforming other related techniques in shipboard UAV maritime object detection.