This repo is based on https://github.com/IGITUGraz/WeatherDiffusion with modification

Datasets

We perform experiments for raindrop removal on the RainDrop datasets.

python train_diffusion.py --config "allweather.yml" --resume 'WeatherDiff1024.pth.tar' --sampling_timesteps 25

python eval_diffusion.py --config "allweather_test.yml" --resume 'patch_ddpm_140000.pth.tar' --test_set 'snow' --sampling_timesteps 25

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
configs		configs
datasets		datasets
models		models
results		results
scratch		scratch
utils		utils
.DS_Store		.DS_Store
LICENSE		LICENSE
README.md		README.md
calculate_psnr_ssim.py		calculate_psnr_ssim.py
cs236_project_final_report_short.pdf		cs236_project_final_report_short.pdf
environment.yml		environment.yml
eval_diffusion.py		eval_diffusion.py
postprocess_outputs.py		postprocess_outputs.py
test.py		test.py
train_diffusion.py		train_diffusion.py