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 条
  • [1] Attentive Generative Adversarial Network for Raindrop Removal from A Single Image
    Qian, Rui
    Tan, Robby T.
    Yang, Wenhan
    Su, Jiajun
    Liu, Jiaying
    2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 2482 - 2491
  • [2] Selective generative adversarial network for raindrop removal from a single image
    Shao, Mingwen
    Li, Le
    Wang, Hong
    Meng, Deyu
    NEUROCOMPUTING, 2021, 426 : 265 - 273
  • [3] Raindrop removal from a single image using a two-step generative adversarial network
    Haiying Xia
    Yang Lan
    Shuxiang Song
    Haisheng Li
    Signal, Image and Video Processing, 2022, 16 : 677 - 684
  • [4] Raindrop removal from a single image using a two-step generative adversarial network
    Xia, Haiying
    Lan, Yang
    Song, Shuxiang
    Li, Haisheng
    SIGNAL IMAGE AND VIDEO PROCESSING, 2022, 16 (03) : 677 - 684
  • [5] Single Image Haze Removal using a Generative Adversarial Network
    Raj, Bharath N.
    Venketeswaran, N.
    2020 INTERNATIONAL CONFERENCE ON WIRELESS COMMUNICATIONS SIGNAL PROCESSING AND NETWORKING (WISPNET), 2020, : 37 - 42
  • [6] Generative Adversarial Network for Image Raindrop Removal of Transmission Line Based on Unmanned Aerial Vehicle Inspection
    Xu, Changbao
    Gao, Jipu
    Wen, Qi
    Wang, Bo
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2021, 2021
  • [7] Enhanced Attentive Generative Adversarial Network for Single-Image Deraining
    Chai, Guoqiang
    Wang, Zhaoba
    Guo, Guodong
    Chen, Youxing
    Jin, Yong
    Wang, Dawei
    Lu, Bin
    Ren, Shilei
    IEEE ACCESS, 2021, 9 : 58390 - 58402
  • [8] Enhanced context encoding for single image raindrop removal
    WANG GuoQing
    YANG Yang
    XU Xing
    LI JingJing
    SHEN HengTao
    Science China Technological Sciences, 2021, (12) : 2640 - 2650
  • [9] Enhanced context encoding for single image raindrop removal
    WANG GuoQing
    YANG Yang
    XU Xing
    LI JingJing
    SHEN HengTao
    Science China(Technological Sciences), 2021, 64 (12) : 2640 - 2650
  • [10] Enhanced context encoding for single image raindrop removal
    Wang GuoQing
    Yang Yang
    Xu Xing
    Li JingJing
    Shen HengTao
    SCIENCE CHINA-TECHNOLOGICAL SCIENCES, 2021, 64 (12) : 2640 - 2650