首页|基于生成对抗网络的水下图像增强方法研究

基于生成对抗网络的水下图像增强方法研究

许凯华

基于生成对抗网络的水下图像增强方法研究

许凯华1
扫码查看

作者信息

  • 1. 燕山大学
  • 折叠

摘要

在海洋资源的开发过程中,水下图像作为人类获取海洋信息的主要手段受到越来越多的重视。水下图像采集与处理技术在海洋探测、水下考古、海洋生物研究等领域扮演着愈发重要的角色。然而,由于水下环境复杂多变,光线在水中传播时受到吸收和散射的干扰,导致采集到的水下图像往往存在颜色失真、对比度降低、细节模糊等问题。这些问题极大地影响了图像信息的准确提取和后续分析。因此,研究水下图像增强技术,提升水下图像质量,具有实用价值和研究意义。本文针对水下图像增强技术研究,提出了适应不同环境需求的水下图像增强方法和多水域环境数据集,具体工作内容如下: (1)本文介绍了传统水下图像增强方法和基于深度学习的水下图像增强方法,阐明了目前水下图像增强任务的难点,对水下图像成像模型、卷积神经网络和生成对抗网络进行深入分析。进行了卷积神经网络和生成对抗网络的水下图像增强对比实验,选取了两个代表性的方法,分别是UWCNN和GAN-RS。通过对实验结果的对比和分析,生成对抗网络在处理图像时展现了更优越的性能且采用无监督的方式具有灵活性,所以选取生成对抗网络作为基本网络框架。 (2)为了解决传统水下图像增强方法处理效果不足的问题,提出了一种基于自注意力机制的水下图像增强算法。同时,考虑到水下图像细节失真问题,在生成器结构中添加了自注意力层,增强提取特征能力并润色细节。进而,针对特征图在经过各个层的计算后会丢失细节信息问题,设计了细节增强模块加入到生成器中,弥补出现的特征损失。最后,针对使用合成图像和真实图像之间固有的领域差距问题,在网络中引入了域自适应机制,使生成图像的真实感更接近空中图像。通过消融实验和与其他算法的比较,所提算法有效的提高了水下图像的质量。 (3)为了解决采用单一水下环境数据集的增强方法不能适用于水下多变环境的问题,提出了一种基于双重约束的水下图像增强方法。针对现有数据集只包含单一水下环境问题,创建了一个模拟多种水域环境的合成水下数据集,增强网络经过该数据集训练能够适应复杂多样的水下环境。考虑到使用生成对抗网络的方法不能得到物理上真实的图像问题,在框架中添加物理模型作为物理约束,结合框架中的循环约束组成双重约束,并设计了双重约束损失函数。最后,针对生成图像不能够保持三维信息问题,提出了传输损失函数,促进了生成图像在三维深度空间与原始图像相似。实验结果表明,与其它图像增强方法相比,所提方法在主观和客观评价上的指标结果最优,能够更好地满足不同水下环境下的需求。

关键词

水下图像增强/生成对抗网络/自注意力机制/双重约束

引用本文复制引用

授予学位

硕士

学科专业

电子信息

导师

李鑫滨/田新华

学位年度

2024

学位授予单位

燕山大学

语种

中文

中图分类号

TP
段落导航相关论文