refusion.yml和ir-sde.yml的区别？

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

521 stars 39 forks source link

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

Open William20234 opened 4 months ago

William20234 commented 4 months ago

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

Algolzw commented 4 months ago

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

EleFreelance commented 2 months ago

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

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

Algolzw commented 2 months ago

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