From coarse to fine: A two stage conditional generative adversarial network for single image rain removal

被引:4
|
作者
Wang, Junsheng [1 ]
Gai, Shan [1 ]
Huang, Xiang [1 ]
Zhang, Hai [1 ]
机构
[1] Nanchang Hangkong Univ, Sch Informat Engn, Nanchang 330063, Jiangxi, Peoples R China
基金
中国国家自然科学基金;
关键词
Rain removal; Two stage; Conditional generative adversarial network; Patch-GAN discriminator; MODEL;
D O I
10.1016/j.dsp.2021.102985
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Images captured in rainy days are often obscured by rain streaks which affect the accuracy of object detection, vehicle and pedestrian recognition. It is hard to restore the texture and color information of the de-rained image by some conventional rain removal algorithms. In order to address the problem, we propose a novel two stage conditional generative adversarial network (TS-CGAN), in which the generator network of the TS-CGAN contains two stage frameworks to better gradually remove rain streaks. In addition, compared with previous neural networks on the rain removal work, patch-GAN discriminator of the TS-CGAN can able to encourage generator to adversely produce satisfactory visual clean images that can solve the artifact problem. Extensive experiments on synthetic and real-world datasets demonstrate that the proposed method significantly outperforms recent state-of-the-art algorithms in terms of qualitative and quantitative measurement. (C) 2021 Elsevier Inc. All rights reserved.
引用
收藏
页数:12
相关论文
共 50 条
  • [31] Single-Image Snow Removal Based on an Attention Mechanism and a Generative Adversarial Network
    Jia, Aiwen
    Jia, Zhen-Hong
    Yang, Jie
    Kasabov, Nikola K.
    [J]. IEEE ACCESS, 2021, 9 : 12852 - 12860
  • [32] Single Image Thin Cloud Removal for Remote Sensing Images Based on Conditional Generative Adversarial Nets
    Zhang, Rui
    Xie, Fengying
    Chen, Jiajie
    [J]. TENTH INTERNATIONAL CONFERENCE ON DIGITAL IMAGE PROCESSING (ICDIP 2018), 2018, 10806
  • [33] SARain-GAN: Spatial Attention Residual UNet Based Conditional Generative Adversarial Network for Rain Streak Removal
    Kolekar, Maheshkumar H.
    Bose, Samprit
    Pai, Abhishek
    [J]. IEEE ACCESS, 2024, 12 : 43874 - 43888
  • [34] Text-to-image generation method based on single stage generative adversarial network
    Yang, Bing
    Na, Wei
    Xiang, Xue-Qin
    [J]. Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2023, 57 (12): : 2412 - 2420
  • [35] Conditional Generative Adversarial Network for Monocular Image Depth Map Prediction
    Hao, Shengang
    Zhang, Li
    Qiu, Kefan
    Zhang, Zheng
    [J]. ELECTRONICS, 2023, 12 (05)
  • [36] Image super-resolution using conditional generative adversarial network
    Qiao, Jiaojiao
    Song, Huihui
    Zhang, Kaihua
    Zhang, Xiaolu
    Liu, Qingshan
    [J]. IET IMAGE PROCESSING, 2019, 13 (14) : 2673 - 2679
  • [37] Learning Image Vegetation Index through a Conditional Generative Adversarial Network
    Suarez, Patricia L.
    Sappa, Angel D.
    Vintimilla, Boris X.
    [J]. 2017 IEEE SECOND ECUADOR TECHNICAL CHAPTERS MEETING (ETCM), 2017,
  • [38] Fiber bundle image restoration using Conditional Generative Adversarial Network
    Xu, Baoteng
    Liu, Jialin
    Zhou, Wei
    Xiong, Daxi
    Yang, Xibin
    [J]. AOPC 2020: DISPLAY TECHNOLOGY; PHOTONIC MEMS, THZ MEMS, AND METAMATERIALS; AND AI IN OPTICS AND PHOTONICS, 2020, 11565
  • [39] Image super-resolution based on conditional generative adversarial network
    Gao, Hongxia
    Chen, Zhanhong
    Huang, Binyang
    Chen, Jiahe
    Li, Zhifu
    [J]. IET IMAGE PROCESSING, 2020, 14 (13) : 3006 - 3013
  • [40] Scale-aware Conditional Generative Adversarial Network for Image Dehazing
    Sharma, Prasen Kumar
    Jain, Priyankar
    Sur, Arijit
    [J]. 2020 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2020, : 2344 - 2354