Accurate and real-time visual detection algorithm for environmental perception of USVS under all-weather conditions

Kaiyuan Dong,Tao Liu,Zhen Shi,Yang Zhang
DOI: https://doi.org/10.1007/s11554-024-01417-9
IF: 2.293
2024-02-23
Journal of Real-Time Image Processing
Abstract:Owing to the intricate and ever-changing nature of the marine environment, traditional marine survey methods are subject to numerous limitations. Unmanned surface vehicles (USVs) have gained significant popularity for their role in automatically identifying and positioning targets in the ocean. To enhance the environmental perception capabilities of USVs in complex marine environments, vision-based sea surface object detection algorithms have emerged as a crucial technological approach. In response to the unique challenges of sea surface object detection tasks and the various extreme situations encountered by USVs in real sea environments, YOLOv7 was chosen as the baseline model due to its excellent trade-off between speed and accuracy. We propose Efficient Multi-Scale Pyramid Attention Networks, which enable easy and rapid multi-scale feature fusion. Additionally, we improve the boundary box loss function. Building upon these optimizations, we develop a new detector called Marit-YOLO, which consistently achieves better accuracy and efficiency than the prior art across a wide spectrum of resource constraints. Marit-YOLO achieved a 5.0% increase in the average precision (AP) value on the independently collected Ocean Buoys Dataset. During generalization validation, the AP value also saw a 7.7% increase on the open-source Singapore Maritime Dataset. The Marit-YOLO algorithm achieves a real-time inference speed of 69 frames per second on a single RTX2080, enabling accurate and real-time target detection in complex sea environments.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?