HeavyRainRemoval icon indicating copy to clipboard operation
HeavyRainRemoval copied to clipboard

Published 20 hours ago • verified publisher icon liruoteng
→

Metadata

An implementation of paper "Heavy Rain Image Restoration: Integrating Physics Model and Conditional Adversarial Learning" (CVPR19)

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

← Metadata

149
Stars
34
Forks
Watchers

Owner

verified publisher icon liruoteng

Metadata

An implementation of paper "Heavy Rain Image Restoration: Integrating Physics Model and Conditional Adversarial Learning" (CVPR19)

Back