RCDNet: A Model-driven Deep Neural Network for Single Image Rain Removal (CVPR2020)

Hong Wang, Qi Xie, Qian Zhao, and Deyu Meng [PDF] [Supplementary Material]

The extension of this work is released as DRCDNet where we propose a dynamic rain kernel inference mechanism.

This is a simple coding framework, which has better compatibility with running environments. The original coding framework for RCDNet CVPR2020 is released at https://github.com/hongwang01/RCDNet

Dependicies

This repository is tested under the following system settings:

Python 3.6

Pytorch 1.4.0

CUDA 10.1

GPU NVIDIA Tesla V100-SMX2

Dataset

Please refer to RCDNet, CVPR2020 for downloading datasets and put them into the corresponding folders according to the folder dictionary in VRGNet, CVPR2021

Citation

@InProceedings{Wang_2020_CVPR,  
author = {Wang, Hong and Xie, Qi and Zhao, Qian and Meng, Deyu},  
title = {A Model-Driven Deep Neural Network for Single Image Rain Removal},  
booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},  
month = {June},  
year = {2020}  
}

Contact

If you have any question, please feel free to concat Hong Wang (Email: [email protected])