一种基于改进暗通道的单幅图像去雾方法技术

本发明专利技术公开了一种基于改进暗通道的单幅图像去雾方法，包括：S1、收集一设定数量的不同场景的原始带雾图像；S2、采用改进后的最大类间方差算法计算出带雾图像的分割阈值；S3、采用暗通道算法对原始带雾图像进行天空域与非天空域的二值分割；S4、对两个二值分割图进行图像相与操作，取交集像素作为天空区域，其余作为非天空域；S5、对步骤S4中得到的粗提取结果进行形态学腐蚀并对其进行导向滤波，以获得天空域与非天空域的精提取结果；S6、计算出大气光值A；S7、得到最终的透射率t；S8、利用大气散射模型生成HSI通道的去雾图像J。本发明专利技术可以判断图像是有无天空区域，实现对各类场景图像去雾。

【技术实现步骤摘要】
一种基于改进暗通道的单幅图像去雾方法
本专利技术涉及图像处理
，更具体地说，特别涉及一种基于改进暗通道的单幅图像去雾方法。
技术介绍
图像去雾算法研究有重要的应用价值和实际意义，在图像获取的过程中如果遇到雾霾天气时，采集设备若受到天气等因素干扰，成像效果会存在一定程度的模糊。在不同的应用领域中，如无人驾驶领域，带雾图像会影响系统作出准确的判断，甚至存在一定程度的安全隐患，这就对图像去雾技术提出了更高的要求。在具有低造价、灵活性高的机载遥感领域中，若数据采集时是雾霾天气，所获取的遥感图像也不能对物体的有效信息进行准确传达，会降低目标识别的能力。正因如此，对于平台获取的数据进行单幅图像去雾的意义是十分重大的。暗通道先验理论是通过对大量户外无雾图像的统计与分析总结得到的，该理论认为在大多数不包括天空的局部区域中，存在一些暗像素在无雾图像中暗像素的值近似为零，而有雾图像中这些暗像素的值不为零且在一定程度上代表了雾的厚度。暗通道的先验理论允许在非天空区域中实现对雾度的准确估计。由于该理论适用于非天空区域，因此应用该先验理论必要的一步是实现对天空区域和非天空区域的准确分割。最大类间方差算法算法(Otsu算法)，是一种自适应的阈值确定算法。在图像分割中，Otsu最佳阈值分割算法对于图像亮度和对比度的作用较小，以类别方差为依据，计算出一个灰度值令类间方差最大或者类内方差最小即为该图像的分割阈值。最大类间方差算法通过灰度值给出最佳的类间分离的阈值，按照图像的灰度特性可以把图像分割为背景部分和目标部分，两者之间的类间方差值越大，表明两者的区别越大；接着通过计算得到一个灰度级最佳阈值来对这两类像素进行分离，使分割错误率最小。因此，有必要开发一种通过对该算法进行阈值改进优化来实现对天空区域和非天空区域的精细化分割，进而实现单幅图像的去雾是非常有必要的。
技术实现思路
本专利技术的目的在于提供一种基于改进暗通道的单幅图像去雾方法，以克服现有技术所存在的缺陷。为了达到上述目的，本专利技术采用的技术方案如下：本专利技术提供一种基于改进暗通道的单幅图像去雾方法，包括以下步骤：S1、收集一设定数量的不同场景的原始带雾图像；S2、采用改进后的最大类间方差算法计算出带雾图像的分割阈值，再对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；S3、采用暗通道算法对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；S4、对步骤S2和步骤S3中获得的两个二值分割图进行图像相与操作，以获得天空域与非天空域的粗提取结果；S5、对步骤S4中得到的粗提取结果进行形态学腐蚀并对其进行导向滤波，以获得天空域与非天空域的精提取结果；S6、利用原始带雾图像和全局暗通道图提取全图大气光值Afull，利用天空域的精提取结果获得原始带雾图像中对应的天空域和暗通道图对应的天空域，再求取出天空域的大气光值Asky，选取天空光值Asky和全图光值Afull中较大的作为实际的大气光值A；S7、使用暗通道图计算非天空域的透射率tno_sky，设定去雾完全系数w，计算天空域的透射率tsky，将非天空域的透射率tno_sky与天空域的透射率tsky合并拼接得到一个完整的透射图，再对拼接后的透射图进行导向滤波得到最终的透射率t；S8、利用大气散射模型生成HSI通道的去雾图像J。进一步地，所述天空区域的判断标准为，若输入带雾图像的像素总个数为N，检测出来的天空区域所包含的像素个数为nsky，天空区域比例定义为P，则当参数P大于设定值时则认为有天空区域；进一步地，所述设定值为8％。进一步地，所述步骤S3中在进行相与操作后，取交集像素作为天空区域，其余作为非天空域。进一步地，所述步骤S8中的大气散射模型的计算公式如下：其中，J(x)为恢复出的去雾图像，A为步骤S6中求取的大气光值，t(x)为步骤S7中求取的透射率，t0是人为设置的参数，用来约束透射率t(x)的下限，使透射率的最小值大于零。与现有技术相比，本专利技术的优点在于：1、本专利技术针对带雾图像的场景多样性可以区分为具有天空区域的带雾图像和不具有天空域的去雾图像，可以满足这两种场景的带雾单幅图像的去雾需求。2、本专利技术考虑到最大类间方差算法对于天空域和非天空域的划分的精细度的局限性，使用改进后的最大类间方差算法对原始带雾图像进行天空域与非天空域的划分，通过对阈值进行修正，确定最佳的分割阈值T，根据分割后的天空区域占输入有雾图像的比例参数来判断图像是有无天空区域，实现对各类场景图像去雾。附图说明为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。图1是本专利技术基于改进暗通道的单幅图像去雾方法的流程图。图2是本专利技术基于改进暗通道的单幅图像去雾系统的框架图。具体实施方式下面结合附图对本专利技术的优选实施例进行详细阐述，以使本专利技术的优点和特征能更易于被本领域技术人员理解，从而对本专利技术的保护范围做出更为清楚明确的界定。参阅图1所示，本专利技术公开了一种基于改进暗通道的单幅图像去雾方法，包括以下步骤：步骤S1、收集一设定数量的不同场景的原始带雾图像，数量可以根据需要进行设定；步骤S2、采用改进后的最大类间方差算法(Otsu)计算出带雾图像的分割阈值，并对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；步骤S3、采用暗通道算法对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；步骤S4、对步骤S2和步骤S3中获得的两个二值分割图进行图像相与操作，取交集像素作为天空区域，其余作为非天空域，以获得天空域与非天空域的粗提取结果；步骤S5、对步骤S4中得到的粗提取结果进行形态学腐蚀并对其进行导向滤波，以获得天空域与非天空域的精提取结果，即：根据分割后的天空区域占输入有雾图像的比例来判断图像是有无天空区域，假设输入带雾图像的像素总个数为N，检测出来的天空区域所包含的像素个数为nsky，天空区域比例定义为P(计算公式如下)，在本实施例中当判断参数P>8％时认为有天空区域：本专利技术与专利CN110148093A一种基于暗通道先验的图像去雾改进方法的差异在于：专利CN110148093A着重于对暗通道图像和亮通道图像及其子图像进行一系列运算等处理，并没有强调天空域与非天空域的划分，本专利技术着重将天空域与非天空域分别处理获得更为精准的大气光值和透射率。本专利技术与专利CN108460735A基于单幅图像的改进暗通道去雾方法的差异在于：专利CN108460735A的步骤1中使用canny算子提取带雾图像的边缘特征，本专利技术为了能够准确区分天空域与非天空域使用带雾图像利用改进后的O本文档来自技高网...

【技术保护点】
1.一种基于改进暗通道的单幅图像去雾方法，其特征在于，包括以下步骤：/nS1、收集一设定数量的不同场景的原始带雾图像；/nS2、采用改进后的最大类间方差算法计算出带雾图像的分割阈值，再对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；/nS3、采用暗通道算法对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；/nS4、对步骤S2和步骤S3中获得的两个二值分割图进行图像相与操作，以获得天空域与非天空域的粗提取结果；/nS5、对步骤S4中得到的粗提取结果进行形态学腐蚀并对其进行导向滤波，以获得天空域与非天空域的精提取结果；/nS6、利用原始带雾图像和全局暗通道图提取全图大气光值Afull，利用天空域的精提取结果获得原始带雾图像中对应的天空域和暗通道图对应的天空域，再求取出天空域的大气光值Asky，选取天空光值Asky和全图光值Afull中较大的作为实际的大气光值A；/nS7、使用暗通道图计算非天空域的透射率t

【技术特征摘要】
1.一种基于改进暗通道的单幅图像去雾方法，其特征在于，包括以下步骤：
S1、收集一设定数量的不同场景的原始带雾图像；
S2、采用改进后的最大类间方差算法计算出带雾图像的分割阈值，再对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；
S3、采用暗通道算法对原始带雾图像进行天空域与非天空域的二值分割以得到二值分割图；
S4、对步骤S2和步骤S3中获得的两个二值分割图进行图像相与操作，以获得天空域与非天空域的粗提取结果；
S5、对步骤S4中得到的粗提取结果进行形态学腐蚀并对其进行导向滤波，以获得天空域与非天空域的精提取结果；
S6、利用原始带雾图像和全局暗通道图提取全图大气光值Afull，利用天空域的精提取结果获得原始带雾图像中对应的天空域和暗通道图对应的天空域，再求取出天空域的大气光值Asky，选取天空光值Asky和全图光值Afull中较大的作为实际的大气光值A；
S7、使用暗通道图计算非天空域的透射率tno_sky，设定去雾完全系数w，计算天空域的透射率tsky，将非天空域的透射率tno_sky与天空域的透射率tsky合并拼接得到一个完整的透射图，...

【专利技术属性】
技术研发人员：付耀文，许淑芳，孙晓镒，杨威，张文鹏，
申请(专利权)人：中国人民解放军国防科技大学，
类型：发明
国别省市：湖南;43