search

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.
https://algolzw.github.io/ir-sde/index.html
MIT License
558 stars 41 forks source link

refusion for inpainting task #70

Closed luwang03 closed 9 months ago

luwang03 commented 9 months ago

Hi. How can i train the refusion model for inpainting task? what should i do to train it?

Algolzw commented 9 months ago

Hi. You may need to retrain a latent-unet model for the inpainting dataset and then change the path (including datasets and pretrained unet weight) in latent-dehazing for inpainting.

luwang03 commented 9 months ago

OK. Thank you very much!