Image Dehazing Based on Generative Adversarial Network
HUANG Shuying1, WANG Bin2, LI Hongxia2, YANG Yong3, HU Wei3
1. School of Computer Science and Technology, Tiangong University, Tianjin 300387 2. School of Software and Internet of Things Engineering, Jiangxi University of Finance and Economics, Nanchang 330013 3. School of Information Management, Jiangxi University of Finance and Economics, Nanchang 330032
Abstract:Compared with the image dehazing methods based on image enhancement or physical model, the current image dehazing methods based on deep learning improve the computational efficiency to a certain extent. Nevertheless, the problems of incomplete dehazing and color distortion still exist in complex scenes. Aiming at the different perceptions of human eyes on global and local features, an algorithm of image dehazing based on generative adversarial networks is proposed. Firstly, a multi-scale generator network is designed. The full-size image and the segmented image block are taken as the input to extract the global contour information and local detail information of the image. Then, a feature fusion module is constructed to fuse the global and local information, and the authenticity of the generated dehazing image is judged by the discriminant network. To make the generated dehazing image closer to the corresponding real haze-free image, a multivariate joint loss function is designed by combining the dark channel prior loss, the adversarial loss, the structural similarity loss and the smooth L1 loss to train the network. Experimental results show that the proposed algorithm is superior to some state-of-the-art dehazing algorithms.
[1] WANG W C, YUAN X H. Recent Advances in Image Dehazing. IEEE/CAA Journal of Automatica Sinica, 2017, 4(3): 410-436. [2] AL-SAMMARAIE M F. Contrast Enhancement of Roads Images with Foggy Scenes Based on Histogram Equalization // Proc of the 10th International Conference on Computer Science and Education. Washington, USA: IEEE, 2015: 95-101. [3] 杨万挺,汪荣贵,方 帅,等.滤波器可变的Retinex雾天图像增强算法.计算机辅助设计与图形学学报, 2010, 22(6): 965-971. (YANG W T, WANG R G, FANG S, et al. Variable Filter Retinex Algorithm for Foggy Image Enhancement. Journal of Computer-Aided Design and Computer Graphics, 2010, 22(6): 965-971.) [4] SEOW M J, ASARI V K. Ratio Rule and Homomorphic Filter for Enhancement of Digital Colour Image. Neurocomputing, 2006, 69(7/8/9): 954-958. [5] JIA J F, YUE H. A Wavelet-Based Approach to Improve Foggy Image Clarity. IFAC Proceedings Volumes, 2014, 47(3): 930-935. [6] TAN R T. Visibility in Bad Weather from a Single Image // Proc of the IEEE Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2008. DOI: 10.1109/CVPR.2008.4587643. [7] HE K M, SUN J, TANG X O. Single Image Haze Removal Using Dark Channel Prior. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011, 33(12): 2341-2353. [8] MENG G F, WANG Y, DUAN J Y, et al. Efficient Image Dehazing with Boundary Constraint and Contextual Regularization // Proc of the IEEE International Conference on Computer Vision. Washington, USA: IEEE, 2013: 617-624. [9] ZHENG M Y, QI G Q, ZHU Z Q, et al. Image Dehazing by an Artificial Image Fusion Method Based on Adaptive Structure Decomposition. IEEE Sensors Journal, 2020, 20(14): 8062-8072. [10] JU M Y, DING C, GUO Y J, et al. IDGCP: Image Dehazing Based on Gamma Correction Prior. IEEE Transactions on Image Proce-ssing, 2019, 29: 3104-3118. [11] CAI B L, XU X M, JIA K, et al. DehazeNet: An End-to-End System for Single Image Haze Removal. IEEE Transactions on Image Processing, 2016, 25(11): 5187-5198. [12] LI B Y, PENG X L, WANG Z Y, et al. AOD-Net: All-in-One Dehazing Network // Proc of the IEEE International Conference on Computer Vision. Washington, USA: IEEE, 2017: 4770-4778. [13] CHEN D D, HE M M, FAN Q N, et al. Gated Context Aggregation Network for Image Dehazing and Deraining // Proc of the IEEE Winter Conference on Applications of Computer Vision. Washington, USA: IEEE, 2019: 1375-1383. [14] LIU Z, XIAO B T, ALRABEIAH M, et al. Single Image Dehazing with a Generic Model-Agnostic Convolutional Neural Network. IEEE Signal Processing Letters, 2019, 26(6): 833-837. [15] QU Y Y, CHEN Y Z, HUANG J Y, et al. Enhanced Pix2pix Dehazing Network // Proc of the IEEE Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2019: 8152-8160. [16] ZHU H Y, XI P, CHANDRASEKHAR V, et al. DehazeGAN: When Image Dehazing Meets Differential Programming // Proc of the 27th International Joint Conference on Artificial Intelligence. New York, USA: ACM, 2018: 1234-1240. [17] ENGIN D, GENC A, EKENEL H K. Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing // Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops. Washington, USA: IEEE, 2018: 838-846. [18] YANG X T, XU Z, LUO J B. Towards Perceptual Image Dehazing by Physics-Based Disentanglement and Adversarial Training[C/OL]. [2021-05-04]. http://www.xyang35.umiacs.io/files/perceptual-image-dehazing.pdf. [19] REN W Q, LIU S, ZHANG H, et al. Single Image Dehazing via Multi-scale Convolutional Neural Networks // Proc of the European Conference on Computer Vision. Berlin, Germany: Springer, 2016: 154-169. [20] ZHANG J, TAO D C. FAMED-Net: A Fast and Accurate Multi-scale End-to-End Dehazing Network. IEEE Transactions on Image Processing, 2020, 29: 72-84. [21] DONG H, PAN J S, XIANG L, et al. Multi-scale Boosted Deha-zing Network with Dense Feature Fusion // Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2020: 2154-2164. [22] YEH C H, HUANG C H, KANG L W. Multi-scale Deep Residual Learning-Based Single Image Haze Removal via Image Decomposition. IEEE Transactions on Image Processing, 2020, 29: 3153-3167. [23] 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. [24] WANG Z, BOVIK A C, SHEIKH H R, et al. Image Quality Assessment: From Error Visibility to Structural Similarity. IEEE Transactions on Image Processing, 2004, 13(4): 600-612. [25] HUYNH-THU Q, GHANBARI M. Scope of Validity of PSNR in Image/Video Quality Assessment. Electronics Letters, 2008, 44(13): 800-801. [26] SHARMA G, WU W C, DALAL E N. The CIEDE2000 Color-Difference Formula: Implementation Notes, Supplementary Test Data, and Mathematical Observations. Color Research and Application, 2005, 30(1): 21-30.