CRDNet

Installation

Framework

1. Python 3.7
2. Pytorch1.0 (with ubuntu 16)
3. Torchvision

Python Dependencies

1. skimage
2. numpy
3. visdom : pip install visdom

Train

Download Weights BaiDuYunLink passwd:ncf3

python3.7 derain.py 

Test

python3.7 derain_test.py
ps:make sure data root is right

Citation:

Cumulative Rain Density Sensing Network for Single Image Derain

    InProceedings{
    author = {L. Peng, A. Jiang, Q. Yi and M. Wang},
    title = {Cumulative Rain Density Sensing Network for Single Image Derain(CRDNet)},
    booktitle = {IEEE Signal Processing Letters(SPL)},
    month = {February},
    year = {2020}
    }
    
    IF it helps you , please quotes us , thanks a lot