Single-image de-raining with a connected multi-stream neural network

被引:0
作者
Pan Y. [1 ]
Shin H. [1 ]
机构
[1] Department of Electrical Engineering, Hanyang University, Ansan
关键词
Convolutional neural network; De-raining; High pass filter; Single-image de-raining;
D O I
10.5573/IEIESPC.2020.9.6.461
中图分类号
学科分类号
摘要
Single-image de-raining is extremely challenging, because rainy images may contain rain streaks with various shapes, and at differing scales and densities. In this paper, we propose a new connected multi-stream neural network for removing rain streaks. In order to better extract rain streaks under different conditions, we use three dense networks with different kernel sizes that can efficiently capture the rain information at different densities. We show that providing useful additional information helps the network to effectively learn about the rain streaks. To guide the removal of rain streaks, we utilize a high pass filter to generate a rain region feature map, which focuses on the structure of the rain streaks and ignores the background in the image. Experiments illustrate that the proposed method significantly improves the removal of rain streaks in both synthetic images and real-world images. Copyrights © 2020 The Institute of Electronics and Information Engineers
引用
收藏
页码:461 / 467
页数:6
相关论文
共 14 条
[11]  
Eigen David, Puhrsch Christian, Fergus Rob, Depth map prediction from a single image using a multi-scale deep network, NIPS, pp. 2366-2374, (2014)
[12]  
Zhang He, Patel Vishal M, Image de-raining using a conditional generative adversarial network, (2017)
[13]  
Wang Zhou, Bovik Alan C, Sheikh Hamid R, Simoncelli Eero P, Image quality assessment: from error visibility to structural similarity, IEEE transactions on image processing, 13, 4, pp. 600-612, (2004)
[14]  
Yasarla Rajeev, Patel Vishal M., Uncertainty Guided Multi-Scale Residual Learning-Using a Cycle Spinning CNN for Single Image De-Raining, Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 8405-8414, (2019)