arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.08388cs.CV

Semi-DerainGAN: A New Semi-supervised Single Image Deraining Network

Hefei University of Technology(合肥工业大学);Harbin Institute of Technology (Shenzhen)(哈尔滨工业大学(深圳));Donghua University(东华大学);City University of Hong Kong(香港城市大学);Zhengzhou University(郑州大学)

Yanyan Wei, Zhao Zhang, Yang Wang, Haijun Zhang, Mingbo Zhao, Mingliang Xu, Meng Wang

英文摘要:

Removing the rain streaks from single image is still a challenging task, since the shapes and directions of rain streaks in the synthetic datasets are very different from real images. Although supervised deep deraining networks have obtained impressive results on synthetic datasets, they still cannot obtain satisfactory results on real images due to weak generalization of rain removal capacity, i.e., the pre-trained models usually cannot handle new shapes and directions that may lead to over-derained/under-derained results. In this paper, we propose a new semi-supervised GAN-based deraining network termed Semi-DerainGAN, which can use both synthetic and real rainy images in a uniform network using two supervised and unsupervised processes. Specifically, a semi-supervised rain streak learner termed SSRML sharing the same parameters of both processes is derived, which makes the real images contribute more rain streak information. To deliver better deraining results, we design a paired discriminator for distinguishing the real pairs from fake pairs. Note that we also contribute a new real-world rainy image dataset Real200 to alleviate the difference between the synthetic and real image do-mains. Extensive results on public datasets show that our model can obtain competitive performance, especially on real images.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑