一种基于暗通道先验的图像去雾改进方法与流程

本发明涉及图像处理技术领域，更具体地，涉及一种基于暗通道先验的图像去雾改进方法。

背景技术：

视频监控技术已经应用于我们生活中，如交通监控和农业生产等室外地点。但是近年来，据气象部门发布的数据，我国部分地区有较长的时间受雾霾天气影响，这干扰了视频监控获取真实场景的图像信息。

而在交通监控方面，管理部门无法通过降质图像获取实时的真实情况，无法完成对交通的调度，大大降低了交通运行能力；另外对于车载视频预警系统，无法对雾天模糊图像做出及时的预警信息，影响驾驶人的判断。对于农业生产，智慧农业依赖植物表型分析系统，对作物的生长做出病害预警和生长预测，但该系统主要通过对植物的生长环境图像和植物叶片图像信息分析，降质模糊图像将会影响处理结果。因此对雾天图像的图像复原方法的研究具有重要意义。

现在对有雾图像的处理主要分为基于图像增强的去雾方法和基于图像复原的去雾方法。基于图像增强的去雾方法是利用已有的成熟图像增强方法对雾天图像进行对比度增强，使得所需部分的图像信息突出，但这类方法没有考虑图像降质的本质原因，会导致其他信息丢失。这类方法常用的有直方图均衡、retinex算法和同态滤波等。而基于图像复原的去雾方法是按照物理模型，对降质图像建模分析，复原出清晰图像。该类方法常用的有基于暗通道先验理论、基于颜色衰减先验和大气光偏振等。暗通道先验的去雾方法是目前复原效果较好，但存在以下缺点：由于对大气光值的估计不准确和透射率天空部分过小，存在天空域的雾天图像处理后有颜色失真问题；另外计算复杂度较高，无法满足实时性。

技术实现要素：

本发明为了克服现有暗通道先验的去雾方法在对天空域的雾天图像处理过程中存在有颜色失真且计算复杂度高，无法满足实时性要求的技术缺陷，提供一种基于暗通道先验的图像去雾改进方法。

为解决上述技术问题，本发明的技术方案如下：

一种基于暗通道先验的图像去雾改进方法，包括以下步骤：

s1：输入雾天图像，得到雾天图像的暗通道图像和亮通道图像；

s2：利用四叉树搜索方法得到大气光估计值，结合亮通道图像和大气光估计值得到大气光图像；

s3：计算得到透射率图像，结合大气光图像对透射率图像进行阈值处理；

s4：对透射率图像进行精细化处理，输出去雾的复原图像。

其中，所述步骤s1具体包括以下步骤：

s11：根据输入的雾天图像计算各像素点的三个颜色通道，将最小值存入最小值通道图像，最大值存入最大值通道图像；

s12：计算最小值通道图像各相邻域像素点与中心点的差值，并与雾天图像标准差比较，若比标准差大则将该点像素值置255并求取该局部区域最小值，并置为中心点像素值，处理后的图像为暗通道图像；

s13：将最大值通道图像进行最大值滤波，得到的图像为亮通道图像。

其中，所述步骤s2具体包括以下步骤：

s21：将暗通道图像裁剪成相同尺寸的子图像，分别计算各子图像的均值和标准差，并分别比较各子图像的均值和标准差的差值，将差值较大的子图像重复步骤s21，直至子图像的像素点个数少于阈值；

s22：在暗通道图像上对应出子图像的区域，分别求出各个区域的三通道像素平均值，将最小值即为大气光估计值；

s23：将亮通道图像与大气光估计值进行计算得到大气光图像a，具体计算公式为：

a＝a*lc+b*a0,(a+b<1)；

其中，lc表示步骤s13求得的亮通道图像；a0为步骤s22求得的大气光估计值。

其中，所述步骤s3具体包括以下步骤：

s31：按照暗通道先验理论的透射率估计公式得到透射率图像；

s32：对大气光图像与暗通道图像做差值，得到差值图像；将差值图像每一个像素值与雾天图像标准差做比较，若比标准差小则将透射率图像对应点的像素值乘以标准差与差值的比值；

s33：将透射率图像进行阈值处理，对透射率图像像素值小于0.1的像素点置0.1，大于0.9的像素点置0.9，。

其中，所述步骤s4具体包括以下步骤：

s41：利用步骤s12得到的暗通道图像作为引导滤波图像，对步骤s33得到的透射率图像进行快速引导滤波，得到细化透射率图像t；

s42：将大气光图像a和细化透射率图像代入大气散射模型中，得到去雾的复原图像。

其中，所述大气散射模型具体计算公式为：

其中，j为去雾后的复原图像，i为输入的雾天图像，a是步骤s23中计算得到的大气光图像，t是步骤s41得到的细化透射率图像。

与现有技术相比，本发明技术方案的有益效果是：

本发明提供一种基于暗通道先验的图像去雾改进方法，通过将四叉树搜索得到的大气光估计值与亮通道图像相结合，得到适应雾天图像的大气光值，避免出现无法存在白色或无法适应天空域的雾天图像问题，且将透射率进行阈值处理，有效改进了天空域部分的透射率过低导致的颜色失真问题提高了图像的亮度和对比度，减少图像的信息熵的丢失；且该方法计算过程简单，具有实时性。

附图说明

图1为本发明的步骤流程示意图；

图2为实施例样本图像示意图；

图3为通过现有暗通道先验的去雾方法得到的结果图像示意图；

图4为本发明方法得到的结果图像示意图。

具体实施方式

附图仅用于示例性说明，不能理解为对本专利的限制；

为了更好说明本实施例，附图某些部件会有省略、放大或缩小，并不代表实际产品的尺寸；

对于本领域技术人员来说，附图中某些公知结构及其说明可能省略是可以理解的。

下面结合附图和实施例对本发明的技术方案做进一步的说明。

实施例1

一种基于暗通道先验的图像去雾改进方法，包括以下步骤：

s1：输入雾天图像，得到雾天图像的暗通道图像和亮通道图像；

s2：利用四叉树搜索方法得到大气光估计值，结合亮通道图像和大气光估计值得到大气光图像；

s3：计算得到透射率图像，结合大气光图像对透射率图像进行阈值处理；

s4：对透射率图像进行精细化处理，输出去雾的复原图像。

更具体的，所述步骤s1具体包括以下步骤：

s11：根据输入的雾天图像计算各像素点的三个颜色通道，将最小值存入最小值通道图像，最大值存入最大值通道图像；

s12：计算最小值通道图像各相邻域像素点与中心点的差值，并与雾天图像标准差比较，若比标准差大则将该点像素值置255并求取该局部区域最小值，并置为中心点像素值，处理后的图像为暗通道图像；

s13：将最大值通道图像进行最大值滤波，得到的图像为亮通道图像。

更具体的，所述步骤s2具体包括以下步骤：

s21：将暗通道图像裁剪成相同尺寸的子图像，分别计算各子图像的均值和标准差，并分别比较各子图像的均值和标准差的差值，将差值较大的子图像重复步骤s21，直至子图像的像素点个数少于阈值；

s22：在暗通道图像上对应出子图像的区域，分别求出各个区域的三通道像素平均值，将最小值即为大气光估计值；

s23：将亮通道图像与大气光估计值进行计算得到大气光图像a，具体计算公式为：

a＝a*lc+b*a0,(a+b<1)；

其中，lc表示步骤s13求得的亮通道图像；a0为步骤s22求得的大气光估计值。

更具体的，所述步骤s3具体包括以下步骤：

s31：按照暗通道先验理论的透射率估计公式得到透射率图像；

s32：对大气光图像与暗通道图像做差值，得到差值图像；将差值图像每一个像素值与雾天图像标准差做比较，若比标准差小则将透射率图像对应点的像素值乘以标准差与差值的比值；

s33：将透射率图像进行阈值处理，对透射率图像像素值小于0.1的像素点置0.1，大于0.9的像素点置0.9，。

更具体的，所述步骤s4具体包括以下步骤：

s41：利用步骤s12得到的暗通道图像作为引导滤波图像，对步骤s33得到的透射率图像进行快速引导滤波，得到细化透射率图像t；

s42：将大气光图像a和细化透射率图像代入大气散射模型中，得到去雾的复原图像。

更具体的，所述大气散射模型具体计算公式为：

其中，j为去雾后的复原图像，i为输入的雾天图像，a是步骤s23中计算得到的大气光图像，t是步骤s41得到的细化透射率图像。

显然，本发明的上述实施例仅仅是为清楚地说明本发明所作的举例，而并非是对本发明的实施方式的限定。对于所属领域的普通技术人员来说，在上述说明的基础上还可以做出其它不同形式的变化或变动。这里无需也无法对所有的实施方式予以穷举。凡在本发明的精神和原则之内所作的任何修改、等同替换和改进等，均应包含在本发明权利要求的保护范围之内。