A simple implementation using Deconv
- architecture.py-contains the architecture of the model used for training
- derain-main.py- calls a saved model of the architecture mentioned in the architecure.py
- point the directory path of test/train cases in derain-main.py for the input to the Derain class.
- python3 derain-main.py
The pretrained model can be found here.