高级检索
刘玉珍, 刘美怡, 林森, 陶志勇. 多尺度特征融合注意力网络的水下图像增强[J]. 计算机辅助设计与图形学学报.
引用本文: 刘玉珍, 刘美怡, 林森, 陶志勇. 多尺度特征融合注意力网络的水下图像增强[J]. 计算机辅助设计与图形学学报.
Underwater Image Enhancement Based on Multi-Scale Feature Fusion and Attention Network[J]. Journal of Computer-Aided Design & Computer Graphics.
Citation: Underwater Image Enhancement Based on Multi-Scale Feature Fusion and Attention Network[J]. Journal of Computer-Aided Design & Computer Graphics.

多尺度特征融合注意力网络的水下图像增强

Underwater Image Enhancement Based on Multi-Scale Feature Fusion and Attention Network

  • 摘要: 摘  要: 水下图像在海洋资源探索中具有重要作用, 针对现有的水下增强方法存在去雾不彻底和细节丢失等问题, 提出一种基于多尺度特征融合注意力网络的水下图像增强方法. 首先, 采用多特征提取模块获取图像特征, 学习不同空间的特征信息, 并通过特征融合模块加强不同空间信息的有效联系, 实现特征的复用和深层次的学习; 然后, 构建特征调制模块, 将低质量信息特征转换为高质量信息特征, 包括通道和像素注意残差块, 将其堆叠成链式结构, 通过动态调制多级特征增强图像细节, 并抑制冗余信息; 最后, 构建包含均方差损失函数、损失函数和感知损失函数的多项式损失函数, 引入异步训练模式提高网络性能. 实验结果表明, 基于EUVP数据集、合成的SUDS数据集和UFO-120数据集, 该方法在主观视觉质量和客观评价指标(UCIQE, NIQE, SURF以及信息熵)上均优于其他经典及新颖方法, 增强后水下图像去雾效果良好, 并且在恢复图像细节方面也具有明显优势, 显著地提高了水下图像的视觉质量.

     

    Abstract: Abstract: Underwater image plays an important role in the exploration of marine resources. Aiming at the problems of incomplete defogging and loss of details in existing underwater enhancement methods, we propose a method based on multi-scale feature fusion attention network. Firstly, the multi feature extraction module extracts image features, learning the different space feature information. Secondly, we use the feature fusion module to strengthen the connection of different spatial information and realize the reuse of features. Thirdly, a feature modulation module is constructed to transform low-quality information features into high-quality ones, including channel and pixel attention residual blocks, which are stacked into a chain structure. Multi-level features are dynamically modulated to enhance image details and suppress redundant information. Finally, constructing a polynomial loss function contains a mean square error loss function, loss function and perceptual loss function. Additionally, the asynchronous training mode is introduced to improve network performance. The comparative experiment shows that based on the EUVP dataset, synthetic SUDS dataset and UFO-120 dataset, the proposed method is superior to other classical and novel methods in subjective visual quality and objective evaluation indicators (UCIQE, NIQE, SURF and information entropy). The enhanced underwater image has an excellent defogging effect and presents conspicuous advantages in restoring image details, which significantly improves the visual quality of the underwater image.

     

/

返回文章
返回