基于关键特征传递的KFT-GAN水下图像增强模型
DOI:
CSTR:
作者:
作者单位:

1.广东海洋大学数学与计算机学院湛江524088;2.广东海洋大学电子与信息工程学院湛江524088; 3.广东省智慧海洋传感网及其装备工程技术研究中心湛江524088

作者简介:

通讯作者:

中图分类号:

TP391;TN919.8

基金项目:

广东省普通高校重点领域新一代信息技术专项(2020ZDZX3008)资助


Underwater image enhancement model based on generative adversarial network with key feature transfer
Author:
Affiliation:

1.College of Mathematics and Computer Science, Guangdong Ocean University, Zhanjiang 524088, China; 2.College of Electronics and Information Engineering, Guangdong Ocean University, Zhanjiang 524088, China; 3. Research Center of Guangdong Smart Oceans Sensor Networks and Equipment Engineering, Zhanjiang 524088, China

Fund Project:

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

    针对水体对光的吸收和散射等现象导致的水下图像出现色偏、模糊和细节丢失等问题,提出了基于关键特征传递的KFT-GAN 水下图像增强模型。设计了 KFT 关键特征传递模块,解决了色彩、边缘和纹理等关键特征在网络中高效传递的问题,并结合深度可分离卷积构建了轻量化的网络模型,与使用普通卷积构建的模型相比,参数量降低了60.6%,增强了模型的学习效率。同时该模块有利于生成器网络的编码阶段从输入图像中获取充足的关键特征,并通过下采样和跳跃连接将提取的关键特征传递至解码阶段进行图像重建,增强了重建图像的质量。另外,基于感知损失原理提出了混合损失函数,同时强调图像的多个关键属性,获得了视觉质量更好的图像效果。模型在EUVP和UIEB数据集上均取得了较好的性能,PSNR值分别为21.384 2和18.025 6,SSIM值分别为0.741 3和0.688 9,通过与传统算法和深度学习算法进行定性和定量的对比实验都证明了该模型的有效性和优越性。

    Abstract:

    To address the challenges of color cast, blurriness, and detail loss in underwater images caused by light absorption and scattering, this study proposes a KFT-GAN for underwater image enhancement. The paper introduces a KFT module that facilitates the efficient transmission of critical features such as color, edges, and texture within the network. By integrating depthwise separable convolution, a lightweight network model is constructed, reducing the parameter count by 60.6% compared to models utilizing conventional convolution, thereby enhancing the model’s learning efficiency. This module enables the generator’s encoding stage to extract essential key features from the input image and transmit these features to the decoding stage through downsampling and skip connections, improving the quality of the reconstructed image. Additionally, this study proposes a hybrid loss function based on perceptual loss principles, emphasizing multiple key attributes of the image to achieve superior visual quality. The proposed model demonstrates excellent performance on both the EUVP and UIEB datasets, achieving PSNR values of 21.384 2 and 18.025 6, and SSIM values of 0.741 3 and 0.688 9, respectively. Qualitative and quantitative comparisons with traditional algorithms and deep learning methods validate the effectiveness and superiority of the proposed model.

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

麦仁贵,王骥.基于关键特征传递的KFT-GAN水下图像增强模型[J].电子测量与仪器学报,2026,40(2):76-85

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-04-30
  • 出版日期:
文章二维码
×
《电子测量与仪器学报》
关于防范虚假编辑部邮件的郑重公告