Single image deraining via deep shared pyramid network

摘要

Single image deraining is a highly ill-posed problem. Existing deep neural network-based algorithms usually use larger deep models to solve this problem, which is less effective and efficient. In this paper, we propose a deep neural network based on feature pyramid to solve image deraining. Our algorithm is motivated that the features at different pyramid levels share similar structures. Based on this property, we develop an effective deep neural network, where the deep models at different feature pyramid levels share the same weight parameters. In addition, we further develop a multi-stream dilation convolution to deal with complex rainy streaks. To preserve the image detail, we develop dense connections that can maintain important features from different levels. Our algorithm is trained in an end-to-end manner. Quantitative and qualitative experimental results demonstrate that the proposed method performs favorably against state-of-the-art deraining methods in terms of accuracy as well as model sizes. The source code and dataset will be available at.