Electronics Optics & Control, Volume. 31, Issue 9, 92(2024)
A YOLOv7 Based Lightweight Underwater Target Detection Algorithm
Underwater target detection is of great significance in marine scienceenvironmental protectionresource developmentmilitary defensecultural heritage protection and other fields.Howeverthe complex underwater environmentpoor underwater image quality and small biological aggregation may lead to missed detection and false detection in underwater target detectionso it is necessary to improve the detection accuracy.The realtime 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 difficultiesan improved network YOLOv7PSS is proposed based on YOLOv7.FirstlyPConv 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.Thenthe SE attention mechanism is added to enhance the feature extraction abilityand SIoU loss function is adopted to accelerate network convergence and optimize model training process.Experimental results show that on the URPC2021 underwater target detection datasetthe proposed algorithm has a mAP of 87.3%which is 7.5% higher than that of the original modeland the parameter quantity is reduced by 11.9%which lays a foundation for the deployment of underwater equipment.
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
Received: Sep. 25, 2023
Accepted: --
Published Online: Oct. 22, 2024
The Author Email: