Laser & Optoelectronics Progress, Volume. 61, Issue 12, 1237009(2024)

Underwater Object Detection Algorithm Integrating Explicit Visual Center and Attention Mechanism

Yang Tao, Bangqian Zhong*, Wenbo Zhao, and Kun Zhou
Author Affiliations
  • School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
  • show less

    In this study, a YOLOv5-based underwater object detection algorithm is proposed to address the challenges of mutual occlusion among underwater marine organisms, low detection accuracy for elongated objects, and presence of numerous small objects in underwater marine biological detection tasks. To redesign the backbone network and improve feature extraction capabilities, the algorithm introduced deformable convolutions, dilated convolutions, and attention mechanisms, mitigating the issues of mutual occlusion and low detection accuracy for elongated objects. Furthermore, a weighted explicit visual center feature pyramid module is proposed to address insufficient feature fusion and reduce the number of failed detections for small objects. Moreover, the network structure of YOLOv5 is adjusted to incorporate a small object detection layer that uses the fused attention mechanism, improving the detection performance for small objects. Experimental results reveal that the improved YOLOv5 algorithm achieves a mean average precision of 87.8% on the URPC dataset, demonstrating a 5.3 percentage points improvement over the original YOLOv5 algorithm while retaining a detection speed of 34 frame/s. The proposed algorithm effectively improves precision and reduces missed and false detection rates in underwater object detection tasks.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yang Tao, Bangqian Zhong, Wenbo Zhao, Kun Zhou. Underwater Object Detection Algorithm Integrating Explicit Visual Center and Attention Mechanism[J]. Laser & Optoelectronics Progress, 2024, 61(12): 1237009

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: Aug. 18, 2023

    Accepted: Sep. 22, 2023

    Published Online: Jun. 5, 2024

    The Author Email: Zhong Bangqian (1831165192@qq.com)

    DOI:10.3788/LOP231947

    CSTR:32186.14.LOP231947

    Topics