基于对比学习的无监督水下图像快速增强
CSTR:
作者:
作者单位:

(1.铜仁职业技术大学 工学院,贵州 铜仁 554300;2.昆明理工大学 信息工程与自动化学院,云南 昆明 650500)

作者简介:

李 恒 (1978-),男,硕士,正高级实验师,硕士生导师,主要从事嵌入式系统、图像处理方面的研究。

通讯作者:

中图分类号:

TP391TN919.8

基金项目:

国家自然科学基金(62263016)和云南省科技厅面上项目(202001AT070038) 资助项目


Rapid unsupervised underwater image enhancement based on contrastive learning
Author:
Affiliation:

(1.College of Engineering, Tongren Polytechnic University, Tongren, Guizhou 554300, China;2.Faculty of Information Engineering and Automation,Kunming University of Science and Technology, Kunming, Yunnan 650500, China)

Fund Project:

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

    目前,基于合成数据集训练的有监督增强模型难以精确模拟水下成像的物理机制,难以适应复杂多变的实际场景;无监督模型多依赖循环一致性框架,结构复杂且效率较低。为此,本文提出了一种基于对比学习的无监督快速水下图像增强算法,通过从浑浊域到清晰域的单向风格转换实现高效增强。算法采用沙漏型瓶颈结构、多特征选择(multi-feature selection,MFS) 和亚像素卷积融合(subpixel convolutional fusion,SCF) 构建生成器,在保障特征提取性能的同时降低复杂度并加速推理速度;优化的对比损失正负样本提取方法提升了模型对多变水下数据的适应性。实验表明,算法的PSNR和UIQM分别提升6.5%和5.3%,推理时间缩短至0.04 s,GPU内存占用减少至少65%,在性能与复杂度间实现了良好平衡。

    Abstract:

    Supervised enhancement models trained on synthetic datasets struggle to accurately simulate the physical mechanisms of underwater imaging,limiting their adaptability to complex real-world scenarios.Unsupervised models,often based on cycle consistency frameworks,tend to suffer from excessive structural complexity and low efficiency.To address these issues,this paper proposes an unsupervised rapid underwater image enhancement algorithm based on contrastive learning,achieving efficient enhancement through unidirectional style transfer from the turbid domain to the clear domain.The algorithm employs a hourglass bottleneck structure with multi-feature selection (MFS) and subpixel convolutional fusion (SCF) to build the generator,ensuring high feature extraction performance while reducing complexity and accelerating inference speed.An optimized contrastive loss with improved positive and negative sampling strategies enhances the model′s adaptability to diverse underwater datasets.Experimental results demonstrate that the proposed algorithm improves PSNR and UIQM by 6.5% and 5.3%,respectively,shortens inference time to 0.04 seconds,and decreases GPU memory by at least 65%,achieving a strong balance between performance and complexity.

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

谭洁,李恒,刘辉,胡雨航.基于对比学习的无监督水下图像快速增强[J].光电子激光,2026,37(3):254~267

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-10-23
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-03-19
  • 出版日期:
文章二维码