search

Easquel / WaveDM

Implementation code for WaveDM: Wavelet-Based Diffusion Models for Image Restoration.
https://ieeexplore.ieee.org/abstract/document/10420512
27 stars 2 forks source link

Reproduction of image denoising #2

Closed IDKiro closed 5 months ago

IDKiro commented 1 year ago

Your work is excellent and it is one of the very few method that have used DM on an image denoising and achieved an advantage over the SOTA models.

So will you release the config files of image denoising?

Zwette commented 7 months ago

Sorry for the late reply. The original account was blocked. The config of image denoising on SIDD is the same as in image raindrop removal, with some differences listed below.

training:
    use_mse: False
    patch_n: 80
    batch_size: 1
    n_epochs: 15000
    n_iters: 2000000
    snapshot_freq: 5000
    validation_freq: 5000