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

Image Dehazing Algorithm Based on Conditional Generation Against Network

为了提高雾天图像的去雾效果,提出了一种基于条件生成对抗网络的图像去雾算法.通过端到端可训练的神经网络对合成的室内和室外数据集进行训练,为了捕捉图像中更多的有用信息,在生成网络中设计了生成器和判别器架构,利用预训练的视觉几何组特征模型和L-1正则化梯度对损失函数进行修正,并在判别器的最后一层引入Sigmoid函数用于特征映射,以便进行概率分析可归一化.利用合成数据集对损失函数进行训练,得到新的损失函数的参数,然后利用室外自然有雾图像数据集对训练得到的新的损失函数进行测试.实验结果表明:所提算法有效解决了去雾图像的颜色失真、过饱和、视觉伪像等问题,生成效果更好的去雾图像.

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 L 1-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.