基于边界约束的煤矿井下尘雾图像去雾算法

Defogging algorithm of underground coal mine dust and fog image based on boundary constraint

  • 摘要: 现有煤矿井下图像去雾算法主要有基于图像增强的去雾算法、基于CNN的去雾算法和基于物理模型的去雾算法。前两者去雾效果差,易出现过曝光。基于物理模型的去雾算法依据大气散射模型对尘雾进行处理,但将基于暗通道的大气光值估计方法应用到煤矿井下环境中,选取的大气光值会较小,易造成图像过曝光、无法抑制点光源照射等问题。针对上述问题,将基于暗原色先验的图像去雾算法(He算法)与基于边界约束及上下文正则化的去雾算法(Meng算法)进行融合,提出了一种基于边界约束的煤矿井下尘雾图像去雾算法。首先,对输入的图像进行伽马校正,对校正后的图像进行颜色通道开运算处理,得到低分辨率的像素块,并从中选取最大亮度值作为煤矿井下的大气光值。其次,分别用He算法与Meng算法对伽马校正后的图像进行处理,对采用Meng算法得到的边界约束图进行引导滤波,得到更为清晰的边界约束图,并将Meng算法与He算法的粗透射率差值进行比较再融合。最后,对融合后的粗透射率进行上下文正则化得到细化透射率,根据得出的大气光值与细化后的透射率,通过大气散射模型得到去雾后的图像。仿真结果表明,基于边界约束的煤矿井下尘雾图像去雾算法没有出现过曝光等问题,且对浓雾图像的去雾效果更好,去雾后的图像也更明亮,颜色更加接近原图。采用峰值信噪比(PSNR)、结构相似性指标(SSIM)、特征相似性指标(FSIM)3种指标对去雾效果进行客观评价,结果表明,提出的算法在PSNR,SSIM,FSIM上相较于He算法平均提升了61.52%,36.51%,24.57%,相较于文献9算法平均提升了15.51%,19.27%,−0.30%,相较于Meng算法平均提升了18.93%,7.19%,1.21%,相较于文献11算法平均提升了18.29%,10.54%,1.19%,说明提出的算法在煤矿井下环境中去雾效果更好、图像更加明亮、细节信息保留更多。

     

    Abstract: The existing defogging algorithms of underground coal mine images mainly include defogging algorithm based on image enhancement, defogging algorithm based on CNN and defogging algorithm based on physical model. The former two have poor defogging effect and are prone to over-exposure. The physical model-based defogging algorithm processes the dust and fog according to the atmospheric scattering model. However, the dark channel-based atmospheric light value estimation method is applied to the underground coal mine environment, and the selected atmospheric light value will be small. The problems of image overexposure, incapability of inhibiting point light source irradiation are easily caused. In order to solve the above problems, the image defogging algorithm based on dark primary color prior (He algorithm) and the defogging algorithm based on boundary constraint and context regularization (Meng algorithm) are fused. The defogging algorithm of underground coal mine dust and fog image based on boundary constraint is proposed. The method comprises the following steps. Firstly, Gamma correction is performed on the input image. The color channel opening operation processing is performed on the corrected image to obtain the low-resolution pixel block. The maximum brightness value is selected from the low-resolution pixel block as the underground atmospheric light value of the coal mine. Secondly, the Gamma-corrected image is processed by He algorithm and Meng algorithm respectively. The boundary constraint map obtained by Meng algorithm is filtered to obtain a clearer boundary constraint map. And the rough transmittance difference of Meng algorithm and He algorithm is compared and then fused. Finally, the contextual regularization is performed on fused rough transmittance to obtain the refined transmittance. The obtained atmospheric light value and the refined transmittance are used to obtain the defogged image through the atmospheric scattering model. The simulation results show that the proposed defogging algorithm of underground coal mine dust and fog image based on boundary constraint has no problems such as overexposure. The defogging effect is better, the defogged image is brighter and the color is closer to the original image. The peak signal-to-noise ratio (PSNR), structural similarity index (SSIM) and feature similarity index (FSIM) are used to objectively evaluate the defogging effect of the proposed algorithm. The results show that the proposed algorithm has an average improvement of 61.52%, 36.51% and 24.57% in PSNR, SSIM and FSIM compared with the He algorithm. Compared with the algorithm in literature 9, the proposed algorithm has increased by 15.51%, 19.27% and −0.30% on average. Compared with Meng algorithm, the proposed algorithm has increased by 18.93%, 7.19% and 1.21% on average. Compared with the algorithm in literature 11, the proposed algorithm has increased by 18.29%, 10.54% and 1.19% on average. It shows that the proposed algorithm has better defogging effect, brighter image and more details in the underground environment of coal mine.

     

/

返回文章
返回