/CRDNet

Cumulative Rain Density Sensing Network for Single Image Derain(CRDNet) IEEE SPL 2020

Primary LanguagePython

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