基于红色通道注意力机制的水下图像增强
CSTR:
作者:
作者单位:

作者简介:

王浩瀚(1999-),男,硕士生,主要从事计算机视觉、图形处理研究

通讯作者:

中图分类号:

TP391

基金项目:

山东省自然科学基金“基于深度时–空特征的视频去模糊方法”(ZR2019MF050)


Underwater Image Enhancement Based on Red Channel Attention Mechanism
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    水下图像增强因其在海洋勘测和水下机器人中的重要意义而备受关注。在过去的几年中,已经提出了许多水下图像增强算法。已有的深度学习方法由于忽略水下图像的预处理过程和对红色通道信息的增强或者弱化了这个过程,导致增强结果并不显著,其往往只适应特定的场景,缺乏泛化能力。为此,基于卷积神经网络建立了一种全新的水下图像增强算法,为了充分利用特征图的通道信息,在相同维度的特征图之间采用不同尺寸的卷积核获取更多通道数目的特征。然后,基于红色通道构建了注意力机制,以加强对于图像中容易丢失信息的红色通道的特征提取。最后,在 EUVP,UFO120 数据集做了消融实验,证明了红色通道注意力机制的有效性。通过对对比实验的增强结果进行各项指标分析,证明增强结果有着更高的结构相似性和峰值信噪比,并且在无参考指标方面有着更高的颜色平衡、清晰度以及对比度,综合性能优于以往的方法。

    Abstract:

    Underwater image enhancement has attracted attention for its importance in marine survey and underwater robotics. In the past few years,many underwater image enhancement algorithms have been proposed. The existing deep learning methods ignore the pre-processing process of underwater image and the enhancement of red channel information or weaken this process,resulting in insignificant enhancement results,which are often adapted to specific scenes only and lack the generalization capability. To this end,we establish a brand new underwater image enhancement algorithm based on convolutional neural network. In order to make full use of the channel information of feature maps,convolutional cores of different dimensions are used to obtain more number of channels in the same dimensions. Then,we construct an attention mechanism based on red channels to enhance the extraction of features for red channels that tend to lose information in images. Finally,we carry out ablation experiments in the EUVP, UFO120 dataset to demonstrate the effectiveness of the red channel attention mechanism. By analyzing the various indicators of the contrast experiment enhancement results,we prove that the enhancement results have higher structural similarity and peak signal-to-noise ratio,and have higher color balance,clarity and contrast in terms of reference-free indicators,and its comprehensive performance is better than the previous methods.

    参考文献
    相似文献
    引证文献
引用本文

王浩瀚,王国栋,张镡月,等.基于红色通道注意力机制的水下图像增强[J].数字海洋与水下攻防,2023,6(1):48-55

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-03-01
  • 出版日期:
文章二维码