Back to Search Start Over

Research on multi-scale fusion image enhancement and improved YOLOv5s lightweight ROV underwater target detection method.

Authors :
Mi, Yongfa
Chi, Mingshan
Zhang, Qiang
Liu, Pengjie
Source :
Scientific Reports. 11/16/2024, Vol. 14 Issue 1, p1-21. 21p.
Publication Year :
2024

Abstract

Underwater target detection technology is very important for Marine resources detection and underwater environment perception. The problems of low-quality underwater images, a large amount of calculation, and low accuracy of target detection models still need to be solved. Firstly, a new underwater image enhancement algorithm based on multi-scale fusion is proposed. The algorithm combines the improved white balance algorithm and the improved CLAHE (Contrast Limited Adaptive Histogram Equalization) and realizes the multi-scale fusion strategy by introducing different weight maps, aiming to improve the visual effect and quality of underwater images. Secondly, to further improve the accuracy and efficiency of underwater target detection, an improved YOLOv5s lightweight underwater target detection algorithm is proposed. The algorithm combines the Ghost convolution module, EMA(Efficient Multi-Scale Attention) mechanism, and CARAFE (Content-Aware ReAssembly of FEatures) up-sampling mode. The comparison on the URPC(Underwater Robot Perception Challenge) public dataset shows that the proposed algorithm has achieved significant improvement in terms of model size, parameter amount, calculation amount, and mAP@0.5. Thirdly, to verify the effectiveness of the image enhancement algorithm and underwater target detection algorithm proposed in this paper, the ROV(Remote Operated Vehicle) is used to conduct experiments in the experimental pool. By designing underwater target detection experiments before and after image enhancement, the results show that the enhanced image significantly improves the detection ability of underwater targets. Finally, underwater target detection software is developed to provide strong support for the application of underwater-related fields. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20452322
Volume :
14
Issue :
1
Database :
Academic Search Index
Journal :
Scientific Reports
Publication Type :
Academic Journal
Accession number :
180935937
Full Text :
https://doi.org/10.1038/s41598-024-79777-2