Awesome
HeavyRainRemoval
0. Installation
Framework
- Python 2.7
- Pytorch (only ubuntu 0.4.0 version is tested so far)
- Torchvision
Python Dependencies
- tqdm
- skimage
- opencv
- numpy
- 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