Dehazing algorithm for adaptively corrected transmission under multi-scale morphology

Abstract

In order to solve the problems of color bias and visual deviation caused by inaccurate estimation of transmittance and atmospheric light in image defogging, a new algorithm based on multi-scale morphological reconstruction with adaptive transmittance and atmospheric light correction was proposed. Firstly, the algorithm used the open operation under morphological reconstruction to replace the minimum filter operation in the dark channel, and used the morphological edge to set the scale of the open operation structure elements, and constructed a multi-scale open operation fusion dark channel. After morphological noise reduction, the exact initial transmittance was obtained. According to the relationship between brightness and saturation difference and transmittance, an adaptive transmittance correction model was fitted with Gaussian function to correct the initial transmittance of the sky fog map. Then the local atmospheric light was improved according to the image brightness information and morphology closure operation. Finally, the proposed algorithm was combined with the atmospheric scattering model to obtain an accurate fog free image. The experimental results showed that the proposed algorithm was suitable for fog image restoration under various scenes, the restoration effect was good, and the brightness was suitable.

Keywords

image dehazing / morphological reconfiguration / multi-scale fusion dark channel / adaptive correction / multi-scene recovery

References

[1] TAN R T. Visibility in bad weather from a single image//2008 IEEE Conference on Computer Vision and Pattern Recognition, June 23-28, 2008, Anchorage, AK, USA. New York: IEEE, 2008: 1-8.

[2] FATTAL R. Single image dehazing. ACM Transactions on Graphics, 2008, 27(3): 1-9.

[3] HE K M, SUN J, TANG X O. Single image haze removal using dark channel prior//2009 IEEE Conference on Computer Vision and Pattern Recognition, June 20-25, 2009, Miami, FL, USA. New York: IEEE, 2009: 1956-1963.

[4] ZHU Q S, MAI J M, SHAO L. A fast single image haze removal algorithm using color attenuation prior. IEEE Transactions on Image Processing: a Publication of the IEEE Signal Processing Society, 2015, 24(11): 3522-3533.

[5] WANG W C, YUAN X H, WU X J, et al. Fast image dehazing method based on linear transformation. IEEE Transactions on Multimedia, 2017, 19(6): 1142-1155.

[6] YANG Y, WANG Z W. Haze removal: push DCP at the edge. IEEE Signal Processing Letters, 2020, 27: 1405-1409.

[7] LI B Y, PENG X L, WANG Z Y, et al. AOD-net: all-in-one dehazing network//2017 IEEE International Conference on Computer Vision (ICCV), October 22-29, 2017, Venice, Italy. New York: IEEE, 2017: 4780-4788.

[8] LI R, PAN J, HE M, et al. Task-oriented network for image dehazing. IEEE Transactions on Image Processing, 2020, 29: 6523-6534.

[9] REN W Q, LIU S, ZHANG H, et al. Single image dehazing via multi-scale convolutional neural networks. International Journal of Computer Vision. 2020, 128(1): 240-259.

[10] ZHENG Z R, REN W Q, CAO X C, et al. Ultra-high-definition image dehazing via multi-guided bilateral learning//2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 20-25, 2021, Nashville, TN, USA. New York: IEEE, 2021: 16180-16189.

[11] LI O, SHUI P L. Noise-robust color edge detection using anisotropic morphological directional derivative matrix. Signal Processing, 2019, 165: 90-103.

[12] LIU X, ZHANG H, TANG Y Y, et al. Scene-adaptive single image dehazing via opening dark channel model. IET Image Processing, 2016, 10(11): 877-884.

[13] SALAZAR-COLORES S, CABAL-YEPEZ E, RAMOS-ARREGUIN J M, et al. A fast image dehazing algorithm using morphological reconstruction. IEEE Transactions on Image Processing, 2019, 28(5): 2357-2366.

[14] JIN X L, ZHANG W, LIU L F. Image defogging algorithm based on guided filtering and adaptive tolerance. Journal on Communications, 2020, 41(5): 27-36.

[15] SUN W, WANG H, SUN C H, et al. Fast single image haze removal via local atmospheric light veil estimation. Computers & Electrical Engineering, 2015, 46: 371-383.

[16] YANG Y, ZHANG H W, ZHANG J L. Single image dehazing combining sky segmentation and transmission mapping. Optics and Precision Engineering, 2021, 29(2): 400-410.

[17] SHEN Y Y, LIU C X, ZHANG J D, et al. Atmospheric light correction and transmission optimization based robust image dehazing. Journal of Computer-Aided Design & Computer Graphics, 2017, 29(9): 1604-1612.

[18] LÜ J W, QIAN F, HAN H N, et al. Nighttime image dehazing with a new light segmentation method and a linear image depth estimation model. Chinese Optics, 2022, 15(1): 34-44.

[19] LI B Y, REN W Q, FU D P, et al. Benchmarking single-image dehazing and beyond. IEEE Transactions on Image Processing, 2019, 28(1): 492-505.

[20] ZHAO X. Single image dehazing using bounded channel difference prior//IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, June 19-25, 2021, Nashville, TN, USA. New York: IEEE, 2021, 727-735.