This is the offifical implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.
📰 2023-06: The first and preliminary version is realeased. Code may not be cleaned thoroughly, so feel free to open an issue if any question.
pip install -r requirements.txt
python train_weather_script.py
python test_weather_script.py
This code is heavily based on PatchDM. Many thanks to the authors!
If you find this repository/work helpful in your research, welcome to cite the paper and give a star.
@article{huang2024wavedm,
title={Wavedm: Wavelet-based diffusion models for image restoration},
author={Huang, Yi and Huang, Jiancheng and Liu, Jianzhuang and Yan, Mingfu and Dong, Yu and Lyu, Jiaxi and Chen, Chaoqi and Chen, Shifeng},
journal={IEEE Transactions on Multimedia},
year={2024},
publisher={IEEE}
}