DU-GAN

Single Image De-Raining With Dual U-net Generative Adversarial Network

Installation:

1. Python 3.6
2. Tensorflow 1.13.1

Demo using model

WnetGANtest.py --test_rain ./TestData/input --test_ground ./TestData/label --mode test --output ./output/ --checkpoint ./model/

Put our test model at file called "model"

(Our test model can be download at https://drive.google.com/file/d/1H7R1WCtD8KRhpzb08B0GaqjkhWNsfqSy/view?usp=sharing)