Home

Awesome

HeavyRainRemoval

0. Installation

Framework

  1. Python 2.7
  2. Pytorch (only ubuntu 0.4.0 version is tested so far)
  3. Torchvision

Python Dependencies

  1. tqdm
  2. skimage
  3. opencv
  4. numpy
  5. DeepGuidedFilter: pip install guided-filter-pytorch

1. Demo

To run rain removal, please first download the pretrained model. Open your terminal and type:

python test.py

2. Training Dataset

You may download the synthetic Outdoor-Rain (7GB) datasets at here.

3. Train (coming soon)

python train.py