TenMiss/WaveDM

WaveDM

Implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.

Requirements

pip install -r requirements.txt

Training

python train_weather_script.py

Testing

python test_weather_script.py

Acknowledgment

This code is heavily based on PatchDM.