基于多通道偏振与双分支Transformer的水下图像复原
作者:
作者单位:

(河海大学信息科学与工程学院 )

作者简介:

许馨月(2000—),女,硕士研究生,主要从事水下图像处理研究。E-mail:hhuxxy@hhu.edu.cn

通讯作者:

中图分类号:

TP391.41

基金项目:

国家重点研发计划项目(2023YFB3907203);国家自然科学基金“叶企孙”科学基金项目(U2441254);中央高校基本科研业务费专项资金项目(B250201257)


Underwater image restoration based on multi-channel polarization and dual-branch Transformer
Author:
Affiliation:

(College of Information Science and Engineering, Hohai University )

Fund Project:

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

    针对水下浑浊环境引起的图像对比度低、色彩失真及细节模糊问题,提出一种基于多通道偏振与双分支Transformer的水下图像复原网络。该网络采用基于Stokes矢量的偏振特征自适应融合模块提取水体偏振物理特征,然后通过构建空间-偏振双分支架构,利用动态门控机制自适应融合偏振强度、偏振度和偏振角等多源信息与空间特征,克服单一模态信息表征不全面的局限,且网络主干采用集成多尺度特征交互、通道注意力及多尺度空间Transformer模块的编解码结构,强化退化表征与全局细节还原能力。实验结果表明,该方法在结构还原、细节保持和色彩校正方面优势显著(SSIM为0.854 0,PSNR为32.93.dB,感知指标LPIPS为0.363 3),能显著提升复原图像的清晰度、对比度和色彩保真度。

    Abstract:

    To address the problems of low image contrast, color distortion, and blurred details caused by turbid underwater environments, an underwater image restoration network based on multi-channel polarization and a dual-branch Transformer was proposed. A polarization feature adaptive fusion module based on the Stokes vector was used to extract the physical polarization features of water. A spatial-polarization dual-branch architecture was constructed, and a dynamic gating mechanism was utilized to adaptively fuse multi-source information, such as polarization intensity, degree of polarization, and polarization angle, with spatial features, overcoming the limitation of incomplete representation by single-modality information. An encoder-decoder structure integrating multi-scale feature interaction, channel attention, and multi-scale spatial Transformer modules was adopted for the network backbone to enhance the capabilities of degradation representation and global detail restoration. The experimental results show that this method has significant advantages in structure restoration, detail preservation, and color correction (SSIM: 0.854 0, PSNR: 32.93 dB, perceptual metric LPIPS: 0.363 3), and can significantly improve the clarity, contrast, and color fidelity of the restored images.

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

许馨月,霍冠英,孔伟丰,等.基于多通道偏振与双分支Transformer的水下图像复原[J].河海大学学报(自然科学版),2026,54(5):49-59.(Xu Xinyue, Huo Guanying, Kong Weifeng, et al. Underwater image restoration based on multi-channel polarization and dual-branch Transformer[J]. Journal of Hohai University (Natural Sciences),2026,54(5):49-59.(in Chinese))

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-07-31
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-10-09
  • 出版日期: