UnfairGAN: An enhanced generative adversarial network for raindrop removal from a single image

被引:7
|
作者
Nguyen, Duc Manh [1 ]
Le, Thao Phuong [2 ]
Vo, Duc My [3 ]
Lee, Sang-Woong [4 ]
机构
[1] Vietnam Acad Sci & Technol, Vietnam Natl Space Ctr, 18 Hoang Quoc Viet, Hanoi 122000, Vietnam
[2] Louisiana State Univ, Dept Biol Sci, Baton Rouge, LA 70803 USA
[3] Univ Utah, Dept Neurol, Frost Lab, 383 Colorow,Room 208, Salt Lake City, UT 84108 USA
[4] Gachon Univ, Pattern Recognit & Machine Learning Lab, 1342 Seongnamdaero, Seongnam 13120, South Korea
关键词
Image deraining; Raindrop removal; Generative adversarial network; Deep Raindrops dataset; WINDSHIELD; WEATHER;
D O I
10.1016/j.eswa.2022.118232
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Image deraining is a new challenging problem in real-world applications, such as autonomous vehicles. In a bad weather condition of heavy rainfall, raindrops, mainly hitting glasses or windshields, can significantly reduce observation ability. Moreover, raindrops spreading over the glass can yield refraction's physical effect, which seriously impedes the sightline or undermine machine learning systems. In this paper, we propose an enhanced generative adversarial network to deal with the challenging problems of raindrops. UnfairGAN is an enhanced generative adversarial network that can utilize prior high-level information, such as edges and rain estimation, to boost deraining performance. UnfairGAN can effectively conserve the essential details caused by heavy raindrops and remove artifacts caused by the instability of training the discriminator. This method is based on three main primary advantages. First, UnfairGAN consists of an advanced loss function of the discriminator that can improve the instabilities of traditional GAN. Second, UnfairGAN comprises a new advanced activation function that is able to increase the learning effectiveness of image classification and reconstruction. Finally, UnfairGAN is basically built on a new end-to-end cascade network, namely DRD-UNet, used to probe hierarchical features for image restoration effectively. When evaluating competing methods on the well-known Raindrop dataset, we achieve a peak signal-to-noise ratio value of 31.56 while retaining the essential details in the image. Besides, we introduce a new large image dataset (DeepRaindrops) for training deep learning networks of removing raindrops. In this dataset, our proposed method is superior to other state-of-the-art approaches of deraining raindrops regarding quantitative metrics and visual quality. Our source codes for UnfairGAN are available at https://github.com/ZeroZero19/UnfairGAN.git.
引用
收藏
页数:12
相关论文
共 50 条
  • [41] Cloud Removal from Aerial Images Using Generative Adversarial Network with Simple Image Enhancement
    Pacot, Mark Phil B.
    Marcos, Nelson
    PROCEEDINGS OF THE 2020 3RD INTERNATIONAL CONFERENCE ON IMAGE AND GRAPHICS PROCESSING (ICIGP 2020), 2020, : 77 - 81
  • [42] A generative adversarial network for image denoising
    Zhong, Yue
    Liu, Lizhuang
    Zhao, Dan
    Li, Hongyang
    MULTIMEDIA TOOLS AND APPLICATIONS, 2020, 79 (23-24) : 16517 - 16529
  • [43] Image Captioning with Generative Adversarial Network
    Amirian, Soheyla
    Rasheed, Khaled
    Taha, Thiab R.
    Arabnia, Hamid R.
    2019 6TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL SCIENCE AND COMPUTATIONAL INTELLIGENCE (CSCI 2019), 2019, : 272 - 275
  • [44] A generative adversarial network for image denoising
    Yue Zhong
    Lizhuang Liu
    Dan Zhao
    Hongyang Li
    Multimedia Tools and Applications, 2020, 79 : 16517 - 16529
  • [45] Multi-Scale Shape Adaptive Network for Raindrop Detection and Removal from a Single Image
    Luo, Hao
    Wu, Qingbo
    Ngan, King Ngi
    Luo, Hanxiao
    Wei, Haoran
    Li, Hongliang
    Meng, Fanman
    Xu, Linfeng
    SENSORS, 2020, 20 (23) : 1 - 18
  • [46] Single Image Defogging Algorithm Based on Conditional Generative Adversarial Network
    Ma, Rui-Qiang
    Shen, Xing-Run
    Zhang, Shan-Jun
    MATHEMATICAL PROBLEMS IN ENGINEERING, 2020, 2020
  • [47] SUPER RESOLUTION FOR SINGLE SATELLITE IMAGE USING A GENERATIVE ADVERSARIAL NETWORK
    Lil, Ran
    Liu, Wangzeng
    Gong, Wenyu
    Zhu, Xiuli
    Wang, Xinpeng
    XXIV ISPRS CONGRESS: IMAGING TODAY, FORESEEING TOMORROW, COMMISSION III, 2022, 5-3 : 591 - 596
  • [48] A lightweight generative adversarial network for single image super-resolution
    Lu, Xinbiao
    Xie, Xupeng
    Ye, Chunlin
    Xing, Hao
    Liu, Zecheng
    Cai, Changchun
    VISUAL COMPUTER, 2024, 40 (01): : 41 - 52
  • [49] Prior guided conditional generative adversarial network for single image dehazing
    Su, Yan Zhao
    Cui, Zhi Gao
    He, Chuan
    Li, Ai Hua
    Wang, Tao
    Cheng, Kun
    NEUROCOMPUTING, 2021, 423 : 620 - 638
  • [50] A lightweight generative adversarial network for single image super-resolution
    Xinbiao Lu
    Xupeng Xie
    Chunlin Ye
    Hao Xing
    Zecheng Liu
    Changchun Cai
    The Visual Computer, 2024, 40 : 41 - 52