Skip to content

Latest commit

 

History

History
35 lines (23 loc) · 721 Bytes

README.md

File metadata and controls

35 lines (23 loc) · 721 Bytes

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