Awesome
WaveDM: Wavelet-Based Diffusion Models for Image Restoration
This is the offifical implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.
Notes
📰 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.
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. Many thanks to the authors!
Citation
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}
}