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

Sci Rep. 2024 Nov 16;14(1):28280. doi: 10.1038/s41598-024-79777-2.

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 [email protected]. 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.