节点文献

基于跨色彩空间Transformer的水下图像增强

Underwater image enhancement based on Transformer across color space

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 宋琳刘飞丁元明姜晨赵维

【Author】 SONG Lin;LIU Fei;DING Yuanming;JIANG Chen;ZHAO Wei;Liaoning Key Laboratory of Communication and Signal Processing;College of Information Engineering, Dalian University;

【通讯作者】 丁元明;

【机构】 辽宁省通信与信号处理重点实验室大连大学信息工程学院

【摘要】 由于水下小颗粒、浮游生物造成光衰减、散射现象,从而导致水下图像存在雾霾效应及模糊,大大影响人眼对水下图像的视觉感知。为了解决这2个问题,提出一个基于跨色彩空间Transformer的水下图像增强模型(简称为Tcolor)。在这个模型中,采用RGB、HSV、LAB三色彩空间U-Net进行特征提取,通过通道传输注意力模块突出最具有区分度的通道和区域。此外,使用跨色彩空间Transformer模块保证增强图像的全局一致性,有效避免伪影的产生,同时在Transformer模块中加入卷积保留局部细节。最后,提出一个联合损失函数进一步提升模型的能力。在2种公开的数据集上进行对比实验,与次优的数据相比,评价指标PSNR提高了6.87%,SSIM提高了6.22%,UIQM提高了2.53%。在主观感知和客观评估中,增强后的图像在细节、对比度和一致性方面均得到了提升。

【Abstract】 Due to the presence of small underwater particles and plankton, which cause light attenuation and scattering phenomenon, which leads to underwater image haze effect and blurring, greatly affecting the visual perception of the human eye on the underwater image. In order to solve these two problems, an underwater image enhancement model based on Transformer across color space(referred to as Tcolor) is proposed. In this model, the three color spaces U-Net of RGB, HSV and LAB are used for feature extraction, and the most discriminative channels and regions are highlighted by the Channel Transformer Attention module. In addition, a cross-color space Transformer module is used to ensure the global consistency of the enhanced image and effectively avoid artifacts, while a convolution is added to the Transformer module to preserve local details. Finally, a joint loss function is proposed to further enhance the capability of the model. Comparison experiments are conducted on 2 publicly available datasets, and compared with the suboptimal data, the evaluation metrics PSNR improves by 6.87%, SSIM improves by 6.22%, and UIQM improves by 2.53%.In both subjective perception and objective evaluation, the enhanced images are improved in terms of detail, contrast and consistency.

【基金】 国家自然科学基金项目(61901079)
  • 【文献出处】 兵器装备工程学报 ,Journal of Ordnance Equipment Engineering , 编辑部邮箱 ,2025年06期
  • 【分类号】TP391.41
  • 【下载频次】45
节点文献中: 

本文链接的文献网络图示:

本文的引文网络