Image Dehazing Based on Channel Attention and Conditional Generative Adversarial Network
CSTR:
Author:
  • Article
  • | |
  • Metrics
  • |
  • Reference [24]
  • |
  • Related [20]
  • |
  • Cited by
  • | |
  • Comments
    Abstract:

    In terms of the problems such as haze residues and color distortion in existing dehazing methods, this study takes advantage of a generative adversarial network in reconstructing image super-resolution and proposes an image dehazing algorithm based on channel attention and conditional generative adversarial network (CGAN-ECA). Specifically, the network is based on the encoder-decoder structure. The generator is designed with the multi-scale residual block (MRBlk) and efficient channel attention (ECA) to expand the receptive field, extract multi-scale features, dynamically adjust the weights of different channels, and improve the utilization rate of features. In addition, the Markovian discriminator (PatchGAN) is used to evaluate images and improve the accuracy in identifying images. At the same time, a content loss is added into the loss function to reduce pixel-level and feature-level losses of dehazing images, retain more image details, and achieve high-quality image dehazing. The test results based on the public dataset RESIDE show that compared with DCP, AOD-Net, DehazeNet, and GCANet models, the proposed model increases the peak signal to noise ratio (PSNR) and the structural similarity index (SSIM) by 36.36% and 8.80%, respectively, and color distortion and haze residue are solved. Therefore, CGAN-ECA is an effective method for image dehazing.

    Reference
    [1] Kim SE, Park TH, Eom IK. Fast single image dehazing using saturation based transmission map estimation. IEEE Transactions on Image Processing, 2020, 29: 1985–1998. [doi: 10.1109/TIP.2019.2948279
    [2] 吴迪, 朱青松. 图像去雾的最新研究进展. 自动化学报, 2015, 41(2): 221–239. [doi: 10.16383/j.aas.2015.c131137
    [3] Pal D, Arora A. Removal of fog effect from highly foggy images using depth estimation and fuzzy contrast enhancement method. Proceedings of the 4th International Conference on Computing Communication and Automation. Greater Noida: IEEE, 2018. 1–6.
    [4] Zarie M, Parsayan A, Hajghassem H. Image contrast enhancement using triple clipped dynamic histogram equalisation based on standard deviation. IET Image Processing, 2019, 13(7): 1081–1089. [doi: 10.1049/iet-ipr.2018.5395
    [5] 高原原, 胡海苗. 基于多子块协同单尺度Retinex的浓雾图像增强. 北京航空航天大学学报, 2019, 45(5): 944–951. [doi: 10.13700/j.bh.1001-5965.2018.0528
    [6] Srinivas K, Bhandari AK, Singh A. Exposure-based energy curve equalization for enhancement of contrast distorted images. IEEE Transactions on Circuits and Systems for Video Technology, 2020, 30(12): 4663–4675. [doi: 10.1109/TCSVT.2019.2960861
    [7] He KM, Sun J, Tang XO. Single image haze removal using dark channel prior. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011, 33(12): 2341–2353. [doi: 10.1109/TPAMI.2010.168
    [8] Li ZG, Zheng JH. Single image de-hazing using globally guided image filtering. IEEE Transactions on Image Processing, 2018, 27(1): 442–450. [doi: 10.1109/TIP.2017.2750418
    [9] 高涛, 刘梦尼, 陈婷, 等. 结合暗亮通道先验的远近景融合去雾算法. 西安交通大学学报, 2021, 55(10): 78–86. [doi: 10.7652/xjtuxb202110009
    [10] 黄文君, 李杰, 齐春. 低秩与字典表达分解的浓雾霾场景图像去雾算法. 西安交通大学学报, 2020, 54(4): 118–125. [doi: 10.7652/xjtuxb202004015
    [11] Pei TY, Ma QY, Xue PF, et al. Nighttime haze removal using bilateral filtering and adaptive dark channel prior. Proceedings of 2019 IEEE 4th International Conference on Image, Vision and Computing. Xiamen: IEEE, 2019. 218–222.
    [12] Cai BL, Xu XM, 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. [doi: 10.1109/TIP.2016.2598681
    [13] Li BY, Peng XL, Wang ZY, et al. AOD-Net: All-in-one dehazing network. Proceedings of 2017 IEEE International Conference on Computer Vision. Venice: IEEE, 2017. 4780–4788.
    [14] Song YF, Li J, Wang XG, et al. Single image dehazing using ranking convolutional neural network. IEEE Transactions on Multimedia, 2018, 20(6): 1548–1560. [doi: 10.1109/TMM.2017.2771472
    [15] Chen DD, He MM, Fan QN, et al. Gated context aggregation network for image dehazing and deraining. Proceedings of 2019 IEEE Winter Conference on Applications of Computer Vision. Waikoloa: IEEE, 2019. 1375–1383.
    [16] Yeh CH, Huang CH, Kang LW. Multi-scale deep residual learning-based single image haze removal via image decomposition. IEEE Transactions on Image Processing, 2019, 29: 3153–3167
    [17] Ullah H, Muhammad K, Irfan M, et al. Light-DehazeNet: A novel lightweight CNN architecture for single image dehazing. IEEE Transactions on Image Processing, 2021, 30: 8968–8982. [doi: 10.1109/TIP.2021.3116790
    [18] 覃宏超, 李炎炎, 龙伟, 等. 基于引导滤波和暗原色先验理论透射率估值的视频去雾算法. 浙江大学学报(工学版), 2018, 52(7): 1302–1309. [doi: 10.3785/j.issn.1008-973X.2018.07.010
    [19] Goodfellow IJ, Pouget-Abadie J, Mirza M, et al. Generative adversarial nets. Proceedings of the 27th International Conference on Neural Information Processing Systems. Montreal: MIT Press, 2014. 2672–2680.
    [20] Mirza M, Osindero S. Conditional generative adversarial nets. arXiv: 1411.1784, 2014.
    [21] He KM, Zhang XY, Ren SQ, et al. Deep residual learning for image recognition. Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016. 770–778.
    [22] Wang QL, Wu BG, Zhu PF, et al. ECA-Net: Efficient channel attention for deep convolutional neural networks. Proceedings of 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020. 11531–11539.
    [23] Isola P, Zhu JY, Zhou TH, et al. Image-to-image translation with conditional adversarial networks. Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition. Honolulu: IEEE, 2017. 5967–5976.
    [24] Li BY, Ren WQ, Fu DP, et al. Benchmarking single-image dehazing and beyond. IEEE Transactions on Image Processing, 2019, 28(1): 492–505. [doi: 10.1109/TIP.2018.2867951
    Cited by
    Comments
    Comments
    分享到微博
    Submit
Get Citation

赵茂军,郭凰,白俊峰,廖聪.基于通道注意力与条件生成对抗网络的图像去雾.计算机系统应用,2022,31(11):167-174

Copy
Share
Article Metrics
  • Abstract:995
  • PDF: 1924
  • HTML: 1936
  • Cited by: 0
History
  • Received:March 03,2022
  • Revised:April 02,2022
  • Online: September 01,2022
Article QR Code
You are the first992285Visitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063