Pre-requisites

The project was developed using python 3 with the following packages.

• Pytorch
• Opencv
• Numpy
• Pytorch Image Models
• Pillow

1. Install Pytorch
2. Install with pip:

pip install -r requirements.txt

Datasets

• Rain 13k - Test: Here
• Place it in datasets

Evaluation

python test.py

or

python test.py --ckp_path <model_path> --in_path <input_folder> --out_path <results_path>