arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.00246cs.CVeess.IV

RAUNE-Net:一种残差与注意力驱动的水下图像增强方法

RAUNE-Net: A Residual and Attention-Driven Underwater Image Enhancement Method

Ocean University of China(中国海洋大学);Qingdao University of Technology(青岛理工大学);Beijing Institute of Technology(北京理工大学)

Wangzhen Peng, Chenghao Zhou, Runze Hu, Jingchao Cao, Yutao Liu

AI总结:

本文提出RAUNE-Net水下图像增强网络,通过瓶颈处高层特征残差学习与下采样中的注意力操作提升鲁棒性和适应性,并构建两个专用数据集验证其在多种真实水下图像上优于八种现有方法。

摘要:

水下图像增强(UIE)由于水下环境的独特性质而面临挑战,包括低对比度、高浑浊度、视觉模糊和颜色失真。近年来,深度学习的应用悄然革新了包括UIE在内的多个科学研究领域。然而,现有的基于深度学习的UIE方法普遍存在鲁棒性弱和适应性有限的问题。本文受残差机制和注意力机制的启发,通过在网络瓶颈处对高层特征进行残差学习,并在下采样过程中引入两个方面的注意力操作,提出了一种更可靠、更合理的UIE网络,称为RAUNE-Net。此外,我们收集并创建了两个专门用于评估UIE方法的数据集,其中包含不同类型的水下失真和退化。实验验证表明,与其他八种UIE方法相比,我们的方法在各种真实世界水下图像上取得了令人鼓舞的客观性能和一致的视觉结果。我们的示例代码和数据集已在 https://github.com/fansuregrin/RAUNE-Net 公开提供。

英文摘要:

Underwater image enhancement (UIE) poses challenges due to distinctive properties of the underwater environment, including low contrast, high turbidity, visual blurriness, and color distortion. In recent years, the application of deep learning has quietly revolutionized various areas of scientific research, including UIE. However, existing deep learning-based UIE methods generally suffer from issues of weak robustness and limited adaptability. In this paper, inspired by residual and attention mechanisms, we propose a more reliable and reasonable UIE network called RAUNE-Net by employing residual learning of high-level features at the network's bottle-neck and two aspects of attention manipulations in the down-sampling procedure. Furthermore, we collect and create two datasets specifically designed for evaluating UIE methods, which contains different types of underwater distortions and degradations. The experimental validation demonstrates that our method obtains promising objective performance and consistent visual results across various real-world underwater images compared to other eight UIE methods. Our example code and datasets are publicly available at https://github.com/fansuregrin/RAUNE-Net.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑