一种基于生成对抗网络的单幅图像去雾方法技术

本发明专利技术提供了一种基于生成对抗网络的单幅图像去雾方法，步骤1，构建生成器网络模型，将经过加雾处理的训练样本集输入到生成器网络模型中，生成模仿测试样本集中无雾图像的初步去雾图像；步骤2，构建判决器网络模型，将初步去雾图像输入到所述判决器网络模型中，计算代价函数，步骤2.1，若代价函数计算结果小于预先设置的去雾阈值，则判断输入图像为测试样本集的无雾图像，并将该生成器网络模型作为最优训练模型；步骤2.2，否则，则判断输入图像为生成器网络模型生成的初步去雾图像，利用tensorflow训练生成对抗网络，转步骤2；步骤3，将有雾图像集输入最优训练模型，输出去雾后的图像。本发明专利技术具有设计科学、实用性强、操作简便和去雾效果高的优点。

A single image de fogging method based on generation of confrontation network

The invention provides a single image defogging method based on generating confrontation network. In step 1, the generator network model is constructed, the training sample set which has been fogged is input into the generator network model, and the preliminary defogging image is generated, which imitates the fog-free image in the test sample set. Input the initial de-fogging image into the network model of the decision maker and calculate the cost function. Step 2.1, if the result of the cost function is less than the pre-set de-fogging threshold, judge the input image as the non-fogging image of the test sample set, and take the generator network model as the optimal training model; Step 2.2, otherwise, judge the input image as the non-fogging image of the test sample set. Judging that the input image is the initial defogging image generated by the generator network model, using tensorflow training to generate the confrontation network, turning to step 2; step 3, the fog image set is input into the optimal training model, and the image after defogging is output. The invention has the advantages of scientific design, strong practicability, simple operation and high fog removal effect.

【技术实现步骤摘要】
一种基于生成对抗网络的单幅图像去雾方法
本专利技术涉及单幅图像去雾
，具体的说，涉及了一种基于生成对抗网络的单幅图像去雾方法。
技术介绍
在雾霾天气下，空气中存在许多大气颗粒。这些颗粒不仅吸收和散射场景的反射光，而且还将一些大气光散射到照相机，导致照相机获取的图像劣化，使得图像对比度低、可见性差，质量严重下降。目前，图像去雾算法主要可以分为三类：第一类是基于图像增强，但是基于图像增强会丢失图像的某些信息特征。第二类是基于物理模型的图像复原，图像复原算法的目的为获得具有良好可见性，同时保持良好的颜色恢复性能的自然清晰的图像；基于模糊条件下图像的劣化原因来建立大气散射的物理模型，首先需要估计物理参数模型，如大气光照强度和透射率(深度)，然后逆解该物理模型来获得无雾图像，但是基于物理模型的图像复原处理范围有限；第三类为基于深度学习的图像去雾算法，例如卷积神经网络应用于图像去雾。为了解决以上存在的问题，人们一直在寻求一种理想的技术解决方案。
技术实现思路
本专利技术的目的是针对现有技术的不足，从而提供一种设计科学、实用性强、操作简便和去雾效果高的基于生成对抗网络的单幅图像去雾方法。为了实现上述目的，本专利技术所采用的技术方案是：一种基于生成对抗网络的单幅图像去雾方法，获取无雾图像集作为测试样本集，对所述无雾图像集利用图像处理软件进行加雾处理得到有雾图像集作为训练样本集，该单幅图像去雾方法还包括，步骤1，构建生成器网络模型，将经过加雾处理的训练样本集输入到所述生成器网络模型中，生成模仿所述测试样本集中无雾图像的初步去雾图像；步骤2，构建判决器网络模型，将所述初步去雾图像输入到所述判决器网络模型中，计算代价函数，步骤2.1，若代价函数计算结果小于预先设置的去雾阈值，则判断输入图像为测试样本集中的无雾图像，并将该生成器网络模型作为最优训练模型；步骤2.2，若代价函数计算结果大于预先设置的去雾阈值，则判断输入图像为生成器网络模型生成的初步去雾图像，利用tensorflow训练生成对抗网络，更新生成器网络模型，转步骤2；步骤3，将训练样本集输入最优训练模型，得到去雾后的图像。基于上述，所述生成器网络模型包括编码器结构和解码器结构，所述编码器结构为八层卷积网络结构，每层卷积网络结构之后设置BatchNormalization层和prelu激活函数；每层卷积网络结构在进行下采样，布幅为2，八层卷积个数分别为64-128-254-512-512-512-512-512；卷积尺寸为4*4，输入图像的尺寸为256*256*3，其中3表示通道数，输出为一维向量；所述解码器结构为八层网络结构，每层依次包括4*4反卷积、BatchNormalization层和prelu激活函数，每层卷积网络结构在进行上采样，卷积尺寸为4*4，解码器结构的八层卷积个数设定为：512-512-512-512-512-254-128-64，解码器每层的最终结果为自身卷积结果与相对称的编码器结构卷积层相加，每层实际的卷积个数为解码器结构与编码器结构的相加：512-1024-1024-1024-1024-512-254-128；输出图像的尺寸为256*256*3。基于上述，所述判决器网络模型包括四层下采样层和一判决层，输入图像尺寸为256*256*3，输出结果为一维；其中，每层下采样层的卷积核大小为4*4，步幅为2，输入图像每经过一层所述下采样层，其长和宽尺寸减少一半，每层所述下采样层依次包括卷积、BatchNormalization层和prelu激活函数，每层下采样层的卷积的个数为64-128-254-512；所述判决层为一维向量，卷积核大小为4*4，步幅为1，卷积个数为1。基于上述，所述代价函数loss计算公式为：loss＝lossGAN+λlossMSE其中，λ表示可调参数，lossGAN表示生成对抗网络代价函数，lossMSE表示图像均方差。基于上述，获取MiddleburyStereoDatasets以及在网上下载明亮的且无雾的图像构成无雾图像集作为测试样本集；利用AdobelightroomCC软件对所述无雾图像集人工加雾，得到有雾图像集作为训练样本集。本专利技术相对现有技术具有突出的实质性特点和显著的进步，具体的说，本专利技术提供了一种基于生成对抗网络的单幅图像去雾方法，将卷积神经网络应用在生成对抗网络GAN上，构建生成器网络模型和判决器网络模型，加雾处理得到的训练样本集作为输入，获取无雾图像集作为测试样本集作为标准参考；本专利技术有效改善了图像质量，其具有设计科学、实用性强、操作简便和去雾效果高的优点。具体实施方式下面通过具体实施方式，对本专利技术的技术方案做进一步的详细描述。一种基于生成对抗网络的单幅图像去雾方法，首先获取无雾图像集作为测试样本集，对所述无雾图像集利用图像处理软件进行加雾处理得到有雾图像集作为训练样本集，该单幅图像去雾方法还包括，步骤1，构建生成器网络模型，将经过加雾处理的训练样本集输入到所述生成器网络模型中，生成模仿所述测试样本集中无雾图像的初步去雾图像；步骤2，构建判决器网络模型，将所述初步去雾图像输入到所述判决器网络模型中，计算代价函数，步骤2.1，若代价函数计算结果小于预先设置的去雾阈值，则判断输入图像为测试样本集的无雾图像，并将该生成器网络模型作为最优训练模型；步骤2.2，若代价函数计算结果大于预先设置的去雾阈值，则判断输入图像为生成器网络模型生成的初步去雾图像，利用tensorflow训练生成对抗网络，更新生成器网络模型，转步骤2；步骤3，将有雾图像集输入最优训练模型，输出去雾后的图像。本专利技术给出了一种获得测试样本集和训练样本集的具体实施方式，获取MiddleburyStereoDatasets以及在网上下载明亮的且无雾的图像，构成无雾图像集；利用AdobelightroomCC软件对所述无雾图像集人工加雾，得到有雾图像集。具体的，所述无雾图像集包含室内室外各种场景，由于现实中很难获得同一场景有雾和无雾的图像对，现有基于学习的图像去雾算法，有雾图像大都是通过深度图经由大气散射模型，随机设置参数人工合成；大气散射模型公式为I＝J·t+A(1-t)；网络输入有雾图像，输出有雾图像透射率，然后再经过逆推计算出无雾图像。为了能够简化上述过程，使得本专利技术的图像去雾方法能够直接经过生成对抗网络得到无雾图像，利用Adobe公司的lightroomCC软件的dehaze功能，人工地为所述无雾图像集进行加雾；同时，为了能够适应不同天气条件下的雾浓度，学习到不同雾浓度图像的特征，对无雾图像集合成了浓度分别为10，20，30，40，50，60，70，80，90，100的雾，得到有雾图像集，作为生成器网络模型的输入；将有雾图像集的图像长和宽裁剪成256*256的大小，以适应判决器网络结构。具体的，所述生成器网络模型包括编码器结构和解码器结构，所述编码器结构为八层卷积网络结构，每层卷积网络结构都在进行下采样，布幅为2；每层卷积网络结构之后有BatchNormalization层和prelu激活函数，提高网络的非线性和泛化能力；卷积尺寸为4*4，输入图像256*256*3的尺寸，其中3表示通道数，最终输出一维向量；八层卷积网络结构的卷积个数分别为64-1本文档来自技高网...

【技术保护点】
1.一种基于生成对抗网络的单幅图像去雾方法，获取无雾图像集作为测试样本集，对所述无雾图像集利用图像处理软件进行加雾处理得到有雾图像集作为训练样本集，其特征在于：该单幅图像去雾方法还包括，步骤1，构建生成器网络模型，将经过加雾处理的训练样本集输入到所述生成器网络模型中，生成模仿所述测试样本集中无雾图像的初步去雾图像；步骤2，构建判决器网络模型，将所述初步去雾图像输入到所述判决器网络模型中，计算代价函数，步骤2.1，若代价函数计算结果小于预先设置的去雾阈值，则判断输入图像为测试样本集中的无雾图像，并将该生成器网络模型作为最优训练模型；步骤2.2，若代价函数计算结果大于预先设置的去雾阈值，则判断输入图像为生成器网络模型生成的初步去雾图像，利用tensorflow训练生成对抗网络，更新生成器网络模型，转步骤2；步骤3，将训练样本集输入最优训练模型，得到去雾后的图像。

【技术特征摘要】
1.一种基于生成对抗网络的单幅图像去雾方法，获取无雾图像集作为测试样本集，对所述无雾图像集利用图像处理软件进行加雾处理得到有雾图像集作为训练样本集，其特征在于：该单幅图像去雾方法还包括，步骤1，构建生成器网络模型，将经过加雾处理的训练样本集输入到所述生成器网络模型中，生成模仿所述测试样本集中无雾图像的初步去雾图像；步骤2，构建判决器网络模型，将所述初步去雾图像输入到所述判决器网络模型中，计算代价函数，步骤2.1，若代价函数计算结果小于预先设置的去雾阈值，则判断输入图像为测试样本集中的无雾图像，并将该生成器网络模型作为最优训练模型；步骤2.2，若代价函数计算结果大于预先设置的去雾阈值，则判断输入图像为生成器网络模型生成的初步去雾图像，利用tensorflow训练生成对抗网络，更新生成器网络模型，转步骤2；步骤3，将训练样本集输入最优训练模型，得到去雾后的图像。2.根据权利要求1所述的基于生成对抗网络的单幅图像去雾方法，其特征在于：所述生成器网络模型包括编码器结构和解码器结构，所述编码器结构为八层卷积网络结构，每层卷积网络结构之后设置BatchNormalization层和prelu激活函数；每层卷积网络结构在进行下采样，布幅为2，八层卷积个数分别为64-128-254-512-512-512-512-512；卷积尺寸为4*4，输入图像的尺寸为256*256*3，其中3表示通道数，输出为一维向量；所述解码器结构为八层网络结构，每层依次包括4*4反卷积、BatchNormalization层和prelu激活函数，每层卷积网络结构在进...

【专利技术属性】
技术研发人员：陈长宝，李德仁，侯长生，郭振强，郧刚，卢建伟，
申请(专利权)人：百年金海科技有限公司，
类型：发明
国别省市：河南,41