[发明专利]图像去雾方法及装置、存储介质、电子设备

说明书

本公开涉及图像处理技术领域，尤其涉及一种图像去雾方法及装置、存储介质、电子设备。该方法可以包括：根据待去雾图像的亮度信息、饱和度信息以及梯度信息分割待去雾图像中的天空区域；根据天空区域内的所有像素计算第一全球大气光成分，并将天空区域中的所有像素的透射率均设置为第一透射率；根据第一全球大气光成分分别计算待去雾图像中的非天空区域中的各像素的第二透射率；根据第一全球大气光成分、第一透射率以及非天空区域中的各像素的第二透射率并结合大气散射模型计算待去雾图像的去雾图像。本公开提高了天空区域分割的准确率，避免了天空区域出现色彩失真的现象，同时也保留了更多的非天空区域的细节信息，使得去雾图像更加清晰自然。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种图像去雾方法及装置、存储介质、电子设备。

背景技术

雾霾天气下，大气中漂浮着大量的悬浮颗粒，这些悬浮颗粒对光具有吸收和散射的作用。因此，在雾天环境，尤其是大雾情况下，可见光成像系统捕捉到的图像出现对比度低、颜色失真、模糊不清等退化现象。这不仅影响了人们的视觉效果，也对各种户外视频监控，如交通监控，带来了严重的安全隐患。因此，如何得到清晰的去雾图像具有重大的应用价值。

目前图像去雾方法大致可以分为两类：一类基于传统的图像增强算，比较有代表性方法的如限制对比度自适应直方图均衡，此类方法不考虑雾图像具体的形成过程，对图像中不同景深区域的增强程度相同。但是没考虑到雾的形成规律，不适用于所有场景，特别是具有较大景深的雾天场景，会出现明显的色彩失真。另一类是基于大气散射模型图像复原方法，例如，基于暗道先验的图像去雾算法，其原理为根据图像的先验信息，通过反推雾霾天气探测器的成像过程，来获取清晰的去雾图像。然而，在图像中存在天空区域时，由于天空区域不符合先验条件，因此，通过此类方法得到的去雾图像中的天空区域会出现色彩失真的现象。此外，目前图像去雾方法还包括利用梯度信息对天空区域进行分割，但是在雾的浓度较大时，通过该方法可能将与天空区域相似的区域错分为天空区域，例如，在雾的浓度较大时，可能将大面积的田地错分为天空区域，因此，仅仅利用梯度信息对天空区域进行分割，达不到理想的效果。另外，目前大部分的去雾方法对淡雾图像的处理效果较好，但对浓雾图像的处理效果不佳，且在场景深度变化比较快的地方会产生明显的halo效应。

需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

发明内容

本公开的目的在于提供一种图像去雾方法及装置、存储介质、电子设备，进而至少在一定程度上克服由于相关技术的限制和缺陷而导致的一个或者多个问题。

根据本公开的一个方面，提供一种图像去雾方法，包括：

根据待去雾图像的亮度信息、饱和度信息以及梯度信息分割所述待去雾图像中的天空区域；

根据所述天空区域内的所有像素计算第一全球大气光成分，并将所述天空区域中的所有像素的透射率均设置为第一透射率；

根据所述第一全球大气光成分分别计算所述待去雾图像中的非天空区域中的各像素的第二透射率；

根据所述第一全球大气光成分、所述第一透射率以及所述非天空区域中的各像素的第二透射率并结合大气散射模型计算所述待去雾图像的去雾图像。

在本公开的一种示例性实施例中，所述根据所述天空区域内的所有像素计算第一全球大气光成分包括：

计算所述天空区域内所有像素分别在R/G/B三个通道上的均值，并据以得到所述第一全球大气光成分。

在本公开的一种示例性实施例中，所述待去雾图像中的非天空区域中的各像素的第二透射率的计算公式为：