光子学报, 2019, 48 (5): 0510002, 网络出版: 2019-06-12   

基于条件生成对抗网络的图像去雾算法

Image Dehazing Algorithm Based on Conditional Generation Against Network
作者单位
江西理工大学 电气工程与自动化学院, 江西 赣州 341000
摘要
为了提高雾天图像的去雾效果, 提出了一种基于条件生成对抗网络的图像去雾算法.通过端到端可训练的神经网络对合成的室内和室外数据集进行训练, 为了捕捉图像中更多的有用信息, 在生成网络中设计了生成器和判别器架构, 利用预训练的视觉几何组特征模型和L1-正则化梯度对损失函数进行修正, 并在判别器的最后一层引入Sigmoid函数用于特征映射, 以便进行概率分析可归一化.利用合成数据集对损失函数进行训练, 得到新的损失函数的参数, 然后利用室外自然有雾图像数据集对训练得到的新的损失函数进行测试.实验结果表明:所提算法有效解决了去雾图像的颜色失真、过饱和、视觉伪像等问题, 生成效果更好的去雾图像.
Abstract
In order to improve the dehazing effect of foggy images, an image defogging algorithm based on conditional generation against network was proposed. Through end-to-end trainable nerves, the network trained the synthesized indoor and outdoor data sets. In order to capture more useful information in the image, the generator and discriminator architecture was designed in the generation network, the loss function was modified using the pre-trained visual geometry group feature model and the L1-regular gradient pair loss. At the last level of the discriminator, the Sigmoid function was applied to the feature map for probabilistic analysis to be normalized. By using the synthetic data set to train the loss function, the parameters of the new loss function were obtained, and then the new trained loss function was tested by the outdoor natural fog image data set. The experimental results show that the algorithm effectively solves the problem of color distortion, oversaturated, and visual artifacts, resulting in a better defogging image.

梁毓明, 张路遥, 卢明建, 杨国亮. 基于条件生成对抗网络的图像去雾算法[J]. 光子学报, 2019, 48(5): 0510002. LIANG Yu-ming, ZHANG Lu-yao, LU Ming-jian, YANG Guo-liang. Image Dehazing Algorithm Based on Conditional Generation Against Network[J]. ACTA PHOTONICA SINICA, 2019, 48(5): 0510002.

本文已被 1 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!