基于改进条件生成对抗网络的水下图像增强方法

摘要：

光在水中传播使得红色光衰减严重,从而导致水下图像在颜色上出现偏绿或偏蓝的颜色色差.针对这一红色光衰减现象,提出了一种改进条件生成对抗网络的方法来对水下图像进行增强处理.首先使用动态阈值对原始图像进行初步颜色校正,之后利用条件生成对抗网络在编码与解码连接处引入链接块学习水下图像与正常图像之间的映射关系来实现水下图像的色彩恢复,再使用双边滤波算法进行图像去噪,提高图像的可视性.在损失函数中引入L1与L2损失结合对图像颜色进行学习,同时加入焦点损失来解决样本比例高度不平衡现象.实验结果表明,此方法在水下图像的颜色失真和模糊方面都有很好的增强效果,获得了较好的视觉效果.

关键词：水下光学图像、图像增强、条件生成对抗网络、残差网络

所属期刊栏目：37

分类号：TP751.1;TP391.4(遥感技术)

资助基金：国家自然科学基金;辽宁省高等学校基本科研项目;辽宁工程技术大学双一流学科创新团队资助项目

在线出版日期：2022-06-16（万方平台首次上网日期，不代表论文的发表时间）

页数：共9页

页码：768-776