├── .gitignore ├── LICENSE ├── README.md ├── calculate_psnr_ssim.py ├── configs ├── allweather.yml ├── allweather128.yml ├── outdoorrain.yml ├── raindrop.yml └── snow100k.yml ├── datasets ├── __init__.py ├── allweather.py ├── outdoorrain.py ├── raindrop.py └── snow100k.py ├── eval_diffusion.py ├── models ├── __init__.py ├── ddm.py ├── restoration.py └── unet.py ├── train_diffusion.py └── utils ├── __init__.py ├── logging.py ├── metrics.py ├── optimize.py └── sampling.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | *__pycache__* 3 | .DS_Store 4 | *.py[cod] 5 | *.xml 6 | venv/ 7 | results/ 8 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/README.md -------------------------------------------------------------------------------- /calculate_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/calculate_psnr_ssim.py -------------------------------------------------------------------------------- /configs/allweather.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/configs/allweather.yml -------------------------------------------------------------------------------- /configs/allweather128.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/configs/allweather128.yml -------------------------------------------------------------------------------- /configs/outdoorrain.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/configs/outdoorrain.yml -------------------------------------------------------------------------------- /configs/raindrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/configs/raindrop.yml -------------------------------------------------------------------------------- /configs/snow100k.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/configs/snow100k.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/allweather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/datasets/allweather.py -------------------------------------------------------------------------------- /datasets/outdoorrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/datasets/outdoorrain.py -------------------------------------------------------------------------------- /datasets/raindrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/datasets/raindrop.py -------------------------------------------------------------------------------- /datasets/snow100k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/datasets/snow100k.py -------------------------------------------------------------------------------- /eval_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/eval_diffusion.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/ddm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/models/ddm.py -------------------------------------------------------------------------------- /models/restoration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/models/restoration.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/models/unet.py -------------------------------------------------------------------------------- /train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/train_diffusion.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/utils/optimize.py -------------------------------------------------------------------------------- /utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IGITUGraz/WeatherDiffusion/HEAD/utils/sampling.py --------------------------------------------------------------------------------