Laser & Optoelectronics Progress, Volume. 61, Issue 24, 2437006(2024)

Lightweight Underwater Target Detection Algorithm Based on Improved YOLOv8n

Guobo Xie, Lihui Liang, Zhiyi Lin*, Songze Lin, and Qing Su
Author Affiliations
  • School of Computer Science and Technology, Guangdong University of Technology, Guangzhou 510006, Guangdong , China
  • show less

    In response to the challenges of fuzzy image and numerous small targets in underwater target detection, which lead to missed detection and false detection with the YOLOv8n algorithm, we proposed an enhanced lightweight underwater target detection algorithm. Initially, within the backbone network, certain convolutions were substituted with non-strided space-to-depth convolution, and a global attention mechanism was introduced to augment global contextual information, thereby improving the network's ability to extract features from blurry and small targets. Subsequently, the conventional upsampling method was replaced with a lightweight upsampling operator, content aware reassembly of features, to broaden the model's receptive field. Furthermore, the normalized Wasserstein distance was introduced and integrated with complete intersection over union to devise a novel localization regression loss function, aimed at increasing the accuracy of small target localization in complex underwater environment. Finally, a dynamic target detection head combined with parameterized rectified linear unit was proposed to enhance the performance of the original detection head, thereby improving the model's proficiency in managing small underwater targets. Experimental results demonstrated that the improved YOLOv8n algorithm achieved a mean average precision of 86.62% on the RUOD dataset, marking a 3.20 percentage points improvement over that of the original YOLOv8n algorithm. The total number of model parameters was 5.67 M, with the number of gigabit floating-point operations is 12.5, fulfilling the criteria for lightweight model.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Guobo Xie, Lihui Liang, Zhiyi Lin, Songze Lin, Qing Su. Lightweight Underwater Target Detection Algorithm Based on Improved YOLOv8n[J]. Laser & Optoelectronics Progress, 2024, 61(24): 2437006

    Download Citation

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

    Category: Digital Image Processing

    Received: Mar. 25, 2024

    Accepted: May. 20, 2024

    Published Online: Dec. 11, 2024

    The Author Email: Zhiyi Lin (lzy291@gdut.edu.cn)

    DOI:10.3788/LOP240955

    CSTR:32186.14.LOP240955

    Topics