基于转换域与自适应伽马校正的去雾算法

计算机工程与科学 ›› 2022, Vol. 44 ›› Issue (09): 1630-1637.

基于转换域与自适应伽马校正的去雾算法

王蓉,杨燕

(兰州交通大学电子与信息工程学院，甘肃兰州 730070)

收稿日期:2021-01-29 修回日期:2021-05-27 接受日期:2022-09-25 出版日期:2022-09-25 发布日期:2022-09-25
基金资助:
国家自然科学基金(61561030)；兰州交通大学教改项目(JG201928)

A dehazing algorithm based on transform domain and adaptive gamma correction

WANG Rong,YANG Yan

(School of Electronic and Information Engineering,Lanzhou Jiaotong University,Lanzhou 730070,China)

Received:2021-01-29 Revised:2021-05-27 Accepted:2022-09-25 Online:2022-09-25 Published:2022-09-25

摘要/Abstract

摘要： 针对暗通道先验算法中存在的光晕效应以及天空区域偏色等问题，提出一种基于转换域与自适应伽马校正的图像去雾算法。首先通过将大气散射模型转换至对数域，结合暗通道先验理论提出对数域正相关关系；再利用高斯函数拟合正相关，从而得到粗级透射率；然后将有雾图像转换至HSV色彩空间，提取亮度分量构造自适应伽马校正因子，对粗级透射率进行修正，并使用交叉双边滤波操作实现透射率的进一步优化；最后结合大气散射模型与改进的局部大气光，实现无雾图像的有效复原。仿真实验表明，与几种经典算法相比，该算法复原结果去雾彻底且细节丰富，具有较好的色彩保真度，更接近真实场景。

关键词: 图像去雾, 暗通道先验, 对数域, 正相关, 自适应校正

Abstract: Aiming at the problems of halo effect and color cast of sky area in dark channel prior algorithm, an image dehazing algorithm based on transform domain and adaptive gamma correction is proposed. By transforming atmospheric scattering model to logarithmic domain, combined with the dark channel prior theory, a positive correlation in logarithmic domain is proposed. Then the Gaussian function is used to fit positive correlation to obtain the coarse transmission. At the same time, the hazy image is converted to HSV color space, the brightness component is extracted to construct an adaptive gamma correction factor, the coarse transmission is corrected, and cross bilateral filtering operation is used to further optimize the transmission. Finally, the restoration of the haze-free image is realized by the atmospheric scattering model and the improved local atmospheric light. Experimental results show that the recovered image has rich details and thorough degree of dehazing in comparison to some classic algorithms. Moreover, it is closer to the real scene because of the better color fidelity.

Key words: image dehazing, dark channel prior, logarithmic domain, positive correlation, adaptive correction

王蓉, 杨燕. 基于转换域与自适应伽马校正的去雾算法[J]. 计算机工程与科学, 2022, 44(09): 1630-1637.

WANG Rong, YANG Yan. A dehazing algorithm based on transform domain and adaptive gamma correction[J]. Computer Engineering & Science, 2022, 44(09): 1630-1637.

参考文献［21］

［1］	Ma Xiao,Shao Li-min,Xu Guan-lei.A clearness algorithm for smog images based on dark channel prior［J］.Computer Engineering & Science,2018,40(12):119-126.(in Chinese)
［2］	Chen Y,Li D,Zhang J Q.Complementary color wavelet:A novel tool for the color image/video analysis and processing［J］.IEEE Transactions on Circuits and Systems for Video Technology,2019,29(1):12-27.
［3］	Wang W C,Yuan X H.Recent advances in image dehazing［J］.IEEE/CAA Journal of Automatica Sinica,2017,4(3):410-436.
［4］	Liu Hai-bo,Yang Jie,Wu Zheng-ping,et al.A fast single image dehazing method based on dark channel prior and Retinex theory［J］.Acta Automatica Sinica,2015,41(7):1264-1273.(in Chinese)
［5］	Cai B L,Xu X M,Jia K,et al.DehazeNet:An end-to-end system for single image haze removal［J］.IEEE Transactions on Image Processing,2016,25(11):5187-5198.
［6］	Ren W Q,Liu S,Zhang H,et al.Single image dehazing via multi-scale convolutional neural networks［C］∥Proc of European Conference on Computer Vision,2016:154-169.
［7］	Li B Y,Peng X L,Wang Z Y,et al.AOD-Net:All-in-one dehazing network［C］∥Proc of IEEE International Conference on Computer Vision,2017:4780-4788.
［8］	Qin X,Wang Z L,Bai Y C,et al.FFA-Net:Feature fusion attention network for single image dehazing［C］∥Proc of Association for the Advancement of Artificial Intelligence (AAAI),2020:11908-11915.
［9］	Tan R T.Visibility in bad weather from a single image［C］∥Proc of IEEE Conference on Computer Vision and Pattern Recognition,2008:1-8.
［10］	Fattal R.Single image dehazing［J］.ACM Transactions on Graphics,2008,27(3):1-9.
［11］	He K M,Sun J,Tang X O.Single image haze removal using dark channel prior［J］.IEEE Transactions on Pattern Analysis and Machine Intelligence,2011,33(12):2341-2353.
［12］	Meng G F,Wang Y,Duan J Y,et al.Efficient image dehazing with boundary constraint and contextual regularization ［C］∥Proc of IEEE International Conference on Computer Vision,2013:617-624.
［13］	Zhu Q S,Mai J M,Shao L.A fast single image haze removal algorithm using color attenuation prior［J］.IEEE Transactions on Image Processing,2015,24(11):3522-3533.
［14］	Narasimhan S G,Nayar S K.Vision and the atmosphere［J］.International Journal of Computer Vision,2002,48(3):233-254.
［15］	Liu Shu-dong,Yao Wen-bo,Zhang Yan.Forest fire monitor- ing based on machine vision in foggy weather［J］.Computer Engineering & Science,2020,42(7):1253-1261.(in Chinese)
［16］	He K M,Sun J,Tang X O.Guided image filtering［J］.IEEE Transactions on Pattern Analysis and Machine Intelligence,2013,35(6):1397-1409.
［17］	Yang Y,Wang Z W,Hong W Q,et al.Single image dehazing algorithm based on double exponential attenuation module［J］.Multimedia Tools and Applications,2021,80(10):15701-15718.
［18］	Sun W,Wang H,Sun C H,et al.Fast single image haze removal via local atmospheric light veil estimation［J］.Computers & Electrical Engineering,2015,46(C):371-378.
［19］	Yang Yan,Wang Zhi-wei.Adaptive image dehazing algorithm based on mean unequal relation optimization［J］.Journal of Electronics & Information Technology,2020,42(3):755-763.(in Chinese)
［20］	Hautiere N, Tarel J P, Aubert D,et al.Blind contrast enhancement assessment by gradient ratioing at visible edges［J］.Image Analysis & Stereology,2011,27(2):87-95.
［21］	Chen Yong,Guo Hong-guang,Ai Ya-peng.Single image dehazing method based on multi-scale convolution neural network［J］.Acta Optica Sinica,2019,39(10):141-150.(in Chinese)
	附中文参考文献：
［1］	马啸,邵利民,徐冠雷.基于暗原色先验的雾霾天气图像清晰化算法［J］.计算机工程与科学,2018,40(12):119-126.
［4］	刘海波,杨杰,吴正平,等.基于暗通道先验和Retinex理论的快速单幅图像去雾算法［J］.自动化学报,2015,41(7):1264-1273.
［15］	刘树东,姚文渤,张艳.雾天条件下基于机器视觉的森林火灾监测［J］.计算机工程与科学,2020,42(7):1253-1261.
［19］	杨燕,王志伟.基于均值不等关系优化的自适应图像去雾算法［J］.电子与信息学报,2020,42(3):755-763.
［21］	陈永,郭红光,艾亚鹏.基于多尺度卷积神经网络的单幅图像去雾方法［J］.光学学报,2019,39(10):141-150.

[1]	唐剑, 车文刚, 高盛祥. 融入注意力机制的多尺度卷积图像去雾方法[J]. 计算机工程与科学, 2023, 45(08): 1453-1462.
[2]	火元莲, 郑海亮, 李明, 张健, . 基于融合亮度模型和梯度域滤波的图像去雾#br#[J]. 计算机工程与科学, 2021, 43(09): 1623-1633.
[3]	许煜，刘辉，尚振宏. 基于多种先验的盲图像复原方法[J]. 计算机工程与科学, 2019, 41(08): 1466-1473.
[4]	杨燕,张宝山,周杰,陈高科. 结合暗通道先验的光补偿快速去雾算法[J]. 计算机工程与科学, 2018, 40(11): 2033-2039.
[5]	刘上力，杨清. 基于改进FP树的项项正相关关联规则挖掘[J]. J4, 2011, 33(7): 183-187.