Image Defogging Algorithm Based on Deblurgan Network
DOI:
https://doi.org/10.54097/fcis.v1i1.1074Keywords:
Single image defogging, Deblurgan, dilation convolution, Spatial attention mechanism, BCE lossAbstract
The reduced visibility in foggy days impairs the quality of captured images and videos to varying degrees, leading to limited applications of these images in the field of computer vision. To solve this problem, direct recovery of fog-free images based on an improved Deblurgan network is proposed. We add the expanded convolution (Dialted Conv) module in the generator to expand the perceptual field to extract richer semantic information, and add the spatial attention mechanism module at specified locations to facilitate the elimination of residual fog; the discriminator uses the traditional PatchGAN for chunk determination to improve the discriminative accuracy; the loss function adds BCE loss to improve the discriminative accuracy of the discriminator and the pixel-level detail retention of the generator. In the synthetic fogged dataset RESIDE, the subjective visual comparison with dark channel, DehazeNet, AOD-Net, and defogging effect shows that the defogging effect of this network model and the detail information and color contrast of defogged images are improved; meanwhile, the objective evaluation indexes Peak Signalto Nise Rtioo a, PSNR and Structure Smilaritiy (SSIM) were also improved respectively.
Downloads
References
KIM T K, PAIK J K, KANG B S. Contrast enhancement system using spatially adaptive histogram equalization with temporal filtering [J]. IEEE Transactions on Consumer Electronics, 1998, 44(1): 82-87.
JOBSON D J, RAHMAN Z, WOODELL G A. A multiscale retinex for bridging the gap between color images and the human observation of scenes[J]. IEEE Transactions on Image Processing, 2002, 6(7): 965-976.
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.
CAI B L, XU X M, JIA K, et al. Dehaze Net: an end-to-endsystem for single image haze removal[J]. IEEE Transactions on Image Processing, 2016, 25(11): 5187-5198.
He Zhang, Vishal M. Patel. Densely Connected Pyramid Dehazing Network [J]. arXiv: 1803.0839 [cs. CV] https://doi.org/10.48550/arXiv.1803.08396
LI B Y, PENG X L, WANG Z Y, et al. AOD-net: All-in-one dehazing network [C]. IEEE International Conference on Computer Vision, 2017: 4770-4778.
MEI K F, JIANG A W, LI J C, et al.Progressive feature fusion network for realistic image dehazing[C]. Asian Conference on Computer Vision, 2018: 203-215.
GOODFELLOW I J, POUGET-ABADIE J, MIRZA M, et al.Generative adversarial nets[C]// Proceedings of the 2014 Neural Information Processing Systems. Cambridge, MA: MIT Press, 2014: 2672-2680.
ZHU H Y, PENG X, CHANDRASEKHAR V, et al. Dehaze GAN: When image dehazing meets diferential programming [C]. The Twenty-Seventh International Joint Conference on Artificial Intelligence, 2018: 1234-1240.
Zhong W F, Zhao J. Image defogging algorithm based on generative adversarial network[J]. Advances in Laser and Optoelectronics 2022,59(4):337-345.
Tu, H Y, Wang W L, Chen J C, et al. A generative adversarial network defogging algorithm combining atmospheric scattering model [J]. Journal of Zhejiang University (Engineering Edition), 2022,56(02),225-235.
Orest Kupyn, Volodymyr Budzan, Mykola Mykhailych, et al. Deblurgan: Blind motion deblurring using conditional adversarial networks[C]. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2018: 8183-8192.
C. Liang-Chieh, G. Papandreou, I. Kokkinos, k. murphy,et al. Semantic Image Segmentation with Deep Convolutional Nets and Fully Connected CRFs. In ICLR, 2015.
YING X Y, WANG Y Q, WANG L G, et al. A stereo attention module for stereo image super-resolution[J]. IEEE Signal Processing Letters, 2020, 27: 496-500.
Gulrajani I, Ahmed, F, Arjovsky, M, et al.Improved Training of Wasserstein GANs[J]. arXiv:1704.00028v3,2017.
De Boer P T, Kroese D P, Mannor S, et al.A tutorial on the cross-entropy method [J]. Annals of operations reserach, 2005, 134 (1): 16-67.
Liu F Y, Shen C H, Lin G S, and Reid L. Learning Depth from Single Monocular Images Using Deep Convolutional Neural Fields[J]. IEEE transactions on pattern analysis and machine intelligence, VOL. 38, NO. 10, 2016.
