一种先验引导的单阶段蒸馏图像去雾方法技术

本发明专利技术公开了一种先验引导的单阶段蒸馏图像去雾方法，包括步骤：一、设计先验引导的单阶段蒸馏去雾网络对单张的有雾图像去雾；二、建立先验引导的单阶段蒸馏去雾网络的损失函数；三、更新权重参数集合；四、调取新的有雾图像循环步骤一至步骤三，直到训练结束，得到先验引导的单阶段蒸馏去雾网络的最终权重参数集合，并确定最终的先验引导的单阶段蒸馏去雾网络；五、单张实际有雾图像去雾。本发明专利技术结合了传统的暗通道去雾算法去雾效果好和端到端去雾算法色彩保真度高的互补优势，在多个数据集的测试中，展现出较强的去雾效果和泛化能力，能够有效提高去雾图像的对比度和视觉效果。能够有效提高去雾图像的对比度和视觉效果。能够有效提高去雾图像的对比度和视觉效果。

【技术实现步骤摘要】
一种先验引导的单阶段蒸馏图像去雾方法


[0001]本专利技术属于单幅图像去雾
，具体涉及一种先验引导的单阶段蒸馏图像去雾方法。

技术介绍

[0002]由于空气中雾、霾等悬浮粒子的影响,光线在传播过程中将发生折射而衰减,导致采集的图像出现颜色偏差、模糊、对比度下降等质量退化现象,这不仅使图像观赏性降低,视觉效果变差,更导致物体特征难以辨认,对拍摄图像的内容理解出现偏差。图像去雾就是利用特定的方法和手段,建立有雾图像和无雾图像之间的特征关系进行去雾处理,进而降低雾霾对图像视觉效果的影响。单幅图像去雾算法是指在仅有一张有雾图像的基础上进行去雾得到清晰图像的过程。
[0003]目前单张图像去雾方法主要分为三大类：第一类是基于图像增强的方法,第二类是基于物理模型的方法,第三类是基于深度学习的方法。
[0004]基于图像增强的方法的本质是对被降质的图像进行增强,改善图像的质量。比如常见的直方均衡、对数变换、幂律变换、锐化、小波变换等。通过这些方法增强图像的对比度或突出图像的特征。与常见的对比度增强方法不同，另一种常见的图像增强的方法是基于颜色恒常性和视网膜皮层理论的Retinex方法。该方法将图像分解为本质图像与照度图像的乘积，从而消除因为受雾霾遮挡的光照因素对图像成像的影响。Retinex方法比传统的对比度提升方法相比，其得到的去雾图像具有更好的局部对比度，颜色失真较小。但由于Retinex方法本身也是一个病态问题，只能进行近似估计，因而也在一定程度上影响了图像去雾效果。
[0005]基于物理模型的方法利用大气散射模型(I＝JT+(1
‑
T)A，其中I表示有雾图像，J表示无雾图像)分别估计场景介质透视率T与全局大气光照A，从而得到清晰的无雾图像。然而在只有单张有雾图像条件下，估计T和A也是一个病态问题，只能进行近视估计。利用大气散射模型将有雾图像恢复到无雾图像的方法，总体上可以分为三类：第1类是基于深度信息的方法；第2类是基于大气光偏振特性的去雾算法；第3类是基于先验知识的方法。前两类方法通常都需要人工配合，才能得到较好的结果，而第3类方法是目前比较常见的方法，比如基于暗通道统计先验的方法，基于颜色统计先验的方法。这些方法由于是通过统计信息得到的知识，并不能适应所有场景，如基于暗通道先验知识的方法对天空等较亮的区域估计透视系统就会出现偏差，导致去雾后的图像整体偏暗。
[0006]基于深度学习的方法通过大样本的合成配对数据集(同一场景下的有雾图像和无雾图像)来训练卷积神经网络，进而建立有雾图像与无雾图像之间的特征关系，具体分为两类：(1)是非端到端的去雾算法，即利用卷积神经网络学习并估计出对应的场景投射图T和全局大气光A，进而通过大气散射模型恢复出无雾图像。这类方法降低了人工先验知识估计参数造成的误差，但由于大气散射模型是一个简化的雾霾生成模型，该方法无法得到高质量的去雾结果；(2)是端到端的去雾算法，即在不估计T和A的基础上，直接将去雾过程视为
图像的变换或者是合成过程。其中，基于图像合成的方法通常利用对比度增强、白平衡、伽玛矫正等方法对有雾图像进行预处理得到衍生图像，然后再通过神经网络学习置信权重，并用学习的权重将衍生图像进行融合，从而实现去雾。这种方法对预处理图像的质量有较高的依赖性，但由于衍生图像只能间接反映无雾场景下的特征，该方法去雾效果有限；基于图像变换的方法直接利用卷积神经网络学习有雾图像与无雾图像之间的映射变换函数，从而得到无雾图像。这种方法通过大样本的数据集拟合了无雾场景与有雾场景之间的特征关系，但由于真实雾霾图像缺乏同一场景下无雾图像的对照，且视觉效果差，这些网络在合成数据集中训练，特征多样性的缺乏导致模型不能很好地去除真实场景中的雾霾。

技术实现思路

[0007]本专利技术所要解决的技术问题在于针对上述现有技术中的不足，提供一种先验引导的单阶段蒸馏图像去雾方法，结合了传统的暗通道去雾算法去雾效果好和端到端去雾算法色彩保真度高的互补优势，在多个数据集的测试中，展现出较强的去雾效果和泛化能力，能够有效提高去雾图像的对比度和视觉效果，便于推广使用。
[0008]为解决上述技术问题，本专利技术采用的技术方案是：一种先验引导的单阶段蒸馏图像去雾方法，其特征在于，该方法包括以下步骤：
[0009]步骤一、设计先验引导的单阶段蒸馏去雾网络对单张的有雾图像去雾，过程如下：
[0010]步骤101、搭建先验引导的重建网络，所述先验引导的重建网络的图像训练样本集为暗通道先验雾天数据集，并将先验引导的重建网络的图像训练样本集中每张有雾图像归一化到固定尺寸；
[0011]步骤102、搭建合成图像去雾网络，所述合成图像去雾网络的图像训练样本集为合成雾天数据集，并将合成图像去雾网络的图像训练样本集中每张有雾图像归一化到固定尺寸；
[0012]步骤103、根据公式Out
FFB
＝G(J
dcp
)
×
α
dcp
+γ(I)
×
α
deh
，构建先验引导的单阶段蒸馏去雾网络Out
FFB
，其中，G(J
dcp
)为先验引导的重建网络，α
dcp
为先验引导的重建网络的权重，γ(I)为合成图像去雾网络，α
deh
为合成图像去雾网络的权重；
[0013]步骤二、建立先验引导的单阶段蒸馏去雾网络的损失函数，过程如下：
[0014]步骤201、根据公式L
dcp
＝||J
dcp
‑
G0(J
dcp
)||，计算先验引导的重建网络的损失函数L
dcp
，其中，J
dcp
为先验引导的重建网络的图像训练样本集中图像利用DCP去雾算法生成的去雾结果，G0(J
dcp
)为先验引导的重建网络的图像训练样本集中图像利用先验引导的重建网络生成的去雾结果，||
·
||为范数运算函数；
[0015]步骤202、根据公式L
deh
＝||J
‑
γ0(I)||，计算合成图像去雾网络的损失函数L
deh
，其中，J为合成图像去雾网络的图像训练样本集中一张清晰无雾图像，γ0(I)为所述清晰无雾图像的有雾图像利用合成图像去雾网络生成的去雾结果；
[0016]步骤203、根据公式计算蒸馏损失L
diss
，其中，为先验引导的单阶段蒸馏去雾网络生成的去雾结果；
[0017]步骤204、根据公式L＝λ1L
dcp
+λ2L
deh
+λ3L
diss
，建立先验引导的单阶段蒸馏去雾网络的损失函数L，其中，λ1为L
dcp
的权重系数，λ2为L
deh
的权重系数，λ3为L
diss
的权重系数；
[0018本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种先验引导的单阶段蒸馏图像去雾方法，其特征在于，该方法包括以下步骤：步骤一、设计先验引导的单阶段蒸馏去雾网络对单张的有雾图像去雾，过程如下：步骤101、搭建先验引导的重建网络，所述先验引导的重建网络的图像训练样本集为暗通道先验雾天数据集，并将先验引导的重建网络的图像训练样本集中每张有雾图像归一化到固定尺寸；步骤102、搭建合成图像去雾网络，所述合成图像去雾网络的图像训练样本集为合成雾天数据集，并将合成图像去雾网络的图像训练样本集中每张有雾图像归一化到固定尺寸；步骤103、根据公式Out
FFB
＝G(J
dcp
)
×
α
dcp
+γ(I)
×
α
deh
，构建先验引导的单阶段蒸馏去雾网络Out
FFB
，其中，G(J
dcp
)为先验引导的重建网络，α
dcp
为先验引导的重建网络的权重，γ(I)为合成图像去雾网络，α
deh
为合成图像去雾网络的权重；步骤二、建立先验引导的单阶段蒸馏去雾网络的损失函数，过程如下：步骤201、根据公式L
dcp
＝||J
dcp
‑
G0(J
dcp
)||，计算先验引导的重建网络的损失函数L
dcp
，其中，J
dcp
为先验引导的重建网络的图像训练样本集中图像利用DCP去雾算法生成的去雾结果，G0(J
dcp
)为先验引导的重建网络的图像训练样本集中图像利用先验引导的重建网络生成的去雾结果，||
·
||为范数运算函数；步骤202、根据公式L
deh
＝||J
‑
γ0(I)||，计算合成图像去雾网络的损失函数L
deh
，其中，J为合成图像去雾网络的图像训练样本集中一张清晰无雾图像，γ0(I)为所述清晰无雾图像的有雾图像利用合成图像去雾网络生成的去雾结果；步骤203、根据公式计算蒸馏损失L
diss
，其中，为先验引导的单阶段蒸馏去雾网络生成的去雾结果；步骤204、根据公式L＝λ1L
dcp
+λ2L
deh
+λ3L
diss
，建立先验引导的单阶段蒸馏去雾网络的损失函数L，...

【专利技术属性】
技术研发人员：王念，崔智高，苏延召，李爱华，兰云伟，蔡艳平，王涛，
申请(专利权)人：中国人民解放军火箭军工程大学，
类型：发明
国别省市：