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 条
  • [21] Unsupervised single image dehazing with generative adversarial network
    Wei Ren
    Li Zhou
    Jie Chen
    Multimedia Systems, 2023, 29 : 2923 - 2933
  • [22] AAGAN: Enhanced Single Image Dehazing With Attention-to-Attention Generative Adversarial Network
    Wang, Wenhui
    Wang, Anna
    Ai, Qing
    Liu, Chen
    Liu, Jinglu
    IEEE ACCESS, 2019, 7 : 173485 - 173498
  • [23] Learning From Synthetic Photorealistic Raindrop for Single Image Raindrop Removal
    Hao, Zhixiang
    You, Shaodi
    Li, Yu
    Li, Kunming
    Lu, Feng
    2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, : 4340 - 4349
  • [24] IMAGE REFLECTION REMOVAL USING THE WASSERSTEIN GENERATIVE ADVERSARIAL NETWORK
    Li, Tingtian
    Lun, Daniel P. K.
    2019 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2019, : 7695 - 7699
  • [25] MARR-GAN: Memristive Attention Recurrent Residual Generative Adversarial Network for Raindrop Removal
    Chai, Qiuyue
    Liu, Yue
    MICROMACHINES, 2024, 15 (02)
  • [26] AI-GAN: Asynchronous interactive generative adversarial network for single image rain removal
    Jin, Xin
    Chen, Zhibo
    Li, Weiping
    PATTERN RECOGNITION, 2020, 100
  • [27] Joint Raindrop and Haze Removal From a Single Image
    Guo, Yina
    Chen, Jianguo
    Ren, Xiaowen
    Wang, Anhong
    Wang, Wenwu
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 9508 - 9519
  • [28] Single Image De-raining Based on a Novel Enhanced Attentive Generative Adversarial Network
    Zhou, Haochen
    Wei, Qin
    5TH ANNUAL INTERNATIONAL CONFERENCE ON INFORMATION SYSTEM AND ARTIFICIAL INTELLIGENCE (ISAI2020), 2020, 1575
  • [29] A physics based generative adversarial network for single image defogging
    Liu, Wei
    Yao, Rongguo
    Qiu, Guoping
    IMAGE AND VISION COMPUTING, 2019, 92
  • [30] Image Specular Highlight Removal using Generative Adversarial Network and Enhanced Grey Wolf Optimization Technique
    Reddy, Maddikera Krishna
    Sekhar, J. C.
    Rao, Vuda Sreenivasa
    Al Ansari, Mohammed Saleh
    El-Ebiary, Yousef A. Baker
    Ramu, Jarubula
    Manikandan, R.
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (06) : 641 - 652