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

多尺度融合图像增强研究及改进的YOLOv5s轻型ROV水下目标检测方法

阅读:3

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.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。