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
580 stars 42 forks source link

refusion.yml和ir-sde.yml的区别? #77

Open William20234 opened 9 months ago

William20234 commented 9 months ago

只是将UNet换成NAFNet,并没有使用[Refusion]这篇文章中提到的训练方法? (不过NAFNet的效果确实好很多)

Algolzw commented 9 months ago

最好还是用refusion的训练方法。不过latent-refusion需要预训练一个unet,这个是专门针对大图输入的可以看情况使用。

EleFreelance commented 6 months ago

最好还是用refusion的训练方法。不过latent-refusion需要预训练一个unet,这个是专门针对大图输入的可以看情况使用。

大佬,好像没有看到关于这个Unent的预训练代码?

Algolzw commented 6 months ago

训练代码在这个unet-latent文件夹里。