High-strength synergic-calibration attention system in YOLO for underwater object detection application

Xin Shen,Guoliang Yuan,Huibing Wang,Xianping Fu
DOI: https://doi.org/10.1007/s00530-024-01545-z
IF: 3.9
2024-11-23
Multimedia Systems
Abstract:Underwater object detection is a prerequisite for underwater robots to achieve accurate perception and efficient operation. However, underwater detection tasks face many inevitable challenges: poor imaging quality, harsh underwater environments and concealed underwater targets. In order to reduce underwater background interference and improve underwater object perception, we explore the optimal attention design suitable for underwater object detection, and propose a high-strength synergic-calibration attention system (HsScAS). The variable information collection stimulates the underwater information extraction potential through average pooling, standard deviation pooling and weighted summation. The channel-wise interaction enhances the underwater attention specificity through sparse channel mapping. The cooperative spatial interaction improve the underwater attention adaptability through dynamic spatial perception and strong spatial action. The integrated information activation stimulates the underwater attention calibration potential through global aggregation attention, detail calibration attention, adaptive fusion attention and semantic selection attention. In order to achieve high-precision and real-time underwater object detection, we add the plug-and-play and cost-effective HsScAS to the one-stage YOLO detector. The effectiveness of our work is fully verified on 5 advanced YOLO detectors, 25 popular attention modules, and 6 different underwater datasets. More importantly, we propose an underwater measurement algorithm, which achieves real-time aquatic size monitoring and stable aquatic density estimation through laser masking, column mapping, distance measurement and output calculation. This is a reliable application of underwater intelligent robots in real marine ranches.
computer science, information systems, theory & methods
What problem does this paper attempt to address?