Electronics Optics & Control, Volume. 31, Issue 9, 92(2024)

A YOLOv7 Based Lightweight Underwater Target Detection Algorithm

TANG Luting and HUANG Hongqiong
Author Affiliations
  • [in Chinese]
  • show less

    Underwater target detection is of great significance in marine scienceenvironmental protectionresource developmentmilitary defensecultural heritage protection and other fields.Howeverthe complex underwater environmentpoor underwater image quality and small biological aggregation may lead to missed detection and false detection in underwater target detectionso it is necessary to improve the detection accuracy.The realtime detection needs to design a faster network structure.Underwater devices have limited storage and computing power and need to maintain low computational overhead while ensuring accuracy.In view of these difficultiesan improved network YOLOv7PSS is proposed based on YOLOv7.FirstlyPConv convolution is used to replace some ordinary convolutions in the backbone network to reduce parameter quantity of the model and speed up training and prediction of the model.Thenthe SE attention mechanism is added to enhance the feature extraction abilityand SIoU loss function is adopted to accelerate network convergence and optimize model training process.Experimental results show that on the URPC2021 underwater target detection datasetthe proposed algorithm has a mAP of 87.3%which is 7.5% higher than that of the original modeland the parameter quantity is reduced by 11.9%which lays a foundation for the deployment of underwater equipment.

    Tools

    Get Citation

    Copy Citation Text

    TANG Luting, HUANG Hongqiong. A YOLOv7 Based Lightweight Underwater Target Detection Algorithm[J]. Electronics Optics & Control, 2024, 31(9): 92

    Download Citation

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

    Received: Sep. 25, 2023

    Accepted: --

    Published Online: Oct. 22, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.09.016

    Topics