改进生成对抗网络水下图像增强方法

• 全文链接
• 请求原文

作   者：

陈海秀;  陆康;  何珊珊;  房威志;  黄仔洁; 

作者机构：

南京信息工程大学自动化学院; 

关键词：

注意力机制;  生成对抗网络;  水下图像;  颜色空间; 

期刊名称：

中国测试

i s s n：

1674-5124

年卷期：

2024 年 50 卷 001 期

页   码：

54-61

摘   要：

针对水下图像颜色失真和细节模糊的问题,提出一种基于改进生成对抗网络的水下图像增强方法.该方法将生成对抗网络作为基础架构,生成网络采用编码解码结构,并引入RGB颜色空间块、HSV颜色空间块和注意力机制;RGB块可以更好地去噪和去除偏色,HSV颜色空间可以调整水下图像的亮度、颜色和饱和度,最后生成网络通过分配权重来生成图像.判别网络采用类似马尔科夫判别器的结构.此外,通过构建全局相似和内容感知多项损失函数,使生成的图像在色彩、内容、结构上和参考图像保持一致.实验表明,所提出的方法在主观比较和客观指标上都有很好的表现.其中结构相似度、峰值信噪比、水下彩色质量评估和水下图像质量度量在合成水下图像测试集的平均值分别为 0.774 6、19.275 8、0.488 9和 3.312 4.在真实水下图像测试集的平均值分别为 0.900 0、24.263 6、0.449 9和3.161 9.在主观评价和客观评价指标上,综合比较,该文算法实验结果均优于对比算法.