liruoteng / Heavyrainremoval
An implementation of paper "Heavy Rain Image Restoration: Integrating Physics Model and Conditional Adversarial Learning" (CVPR19)
Stars: ✭ 93
Programming Languages
python
139335 projects - #7 most used programming language
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
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].