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 条
  • [31] Single image dehazing algorithm based on generative adversarial network
    Zhao, Donghui
    Mo, Bo
    ACM International Conference Proceeding Series, 2022, : 27 - 31
  • [32] Single Image Dehazing via Conditional Generative Adversarial Network
    Li, Runde
    Pan, Jinshan
    Li, Zechao
    Tang, Jinhui
    2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 8202 - 8211
  • [33] Stereoscopic image reflection removal based on Wasserstein Generative Adversarial Network
    Wang, Xiuyuan
    Pan, Yikun
    Lun, Daniel P. K.
    2020 IEEE INTERNATIONAL CONFERENCE ON VISUAL COMMUNICATIONS AND IMAGE PROCESSING (VCIP), 2020, : 148 - 151
  • [34] Image Motion Blur Removal Algorithm Based on Generative Adversarial Network
    Kim, Jongchol
    Kim, Myongchol
    Kim, Insong
    Han, Gyongwon
    Jong, Myonghak
    Ri, Gwuangwon
    PROGRAMMING AND COMPUTER SOFTWARE, 2024, 50 (05) : 403 - 415
  • [35] Adherent mist and raindrop removal from a single image using attentive convolutional network
    He, Da
    Shang, Xiaoyu
    Luo, Jiajia
    NEUROCOMPUTING, 2022, 505 : 178 - 187
  • [36] Single Image Snow Removal via Composition Generative Adversarial networks
    Li, Zhi
    Zhang, Juan
    Fang, Zhijun
    Huang, Bo
    Jiang, Xiaoyan
    Gao, Yongbin
    Hwang, Jenq-Neng
    IEEE ACCESS, 2019, 7 : 25016 - 25025
  • [37] Enhanced Pathology Image Quality with Restore-Generative Adversarial Network
    Rong, Ruichen
    Wang, Shidan
    Zhang, Xinyi
    Wen, Zhuoyu
    Cheng, Xian
    Jia, Liwei
    Yang, Donghan M.
    Xie, Yang
    Zhan, Xiaowei
    Xiao, Guanghua
    AMERICAN JOURNAL OF PATHOLOGY, 2023, 193 (04): : 404 - 416
  • [38] Underwater Image Enhancement Based on Hybrid Enhanced Generative Adversarial Network
    Xu, Danmi
    Zhou, Jiajia
    Liu, Yang
    Min, Xuyu
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2023, 11 (09)
  • [39] A single image raindrop removal model based on lightweight neural network
    Chen Z.
    Bi X.
    Harbin Gongcheng Daxue Xuebao/Journal of Harbin Engineering University, 2023, 44 (02): : 292 - 299
  • [40] Hyperspectral reconstruction from a single textile RGB image based on the generative adversarial network
    Liu, Yue
    Zhang, Jianxin
    Zhang, Yupeng
    TEXTILE RESEARCH JOURNAL, 2023, 93 (1-2) : 307 - 316