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
587 stars 43 forks source link

About IR-SDE pretrained models question... #40

Open AlphaNext opened 1 year ago

AlphaNext commented 1 year ago

感谢您的工作，请问提供的模型文件中，pretrained_models下的模型是在哪个数据集上预训练的？

~~ 另外每个task预训练用到的数据一样吗？如果不一样，用的数据分别是什么？

最后有个小请求，请问可以在百度云盘上分享一份模型文件吗？Google云盘下载有点困难，感谢~~

Algolzw commented 1 year ago

各个任务使用的是不同的数据集：对于去模糊我们用的是GoPro dataset，去雨用的是Rain100L和Rain100H，降噪用的是Flickr和DIV2K等混合的数据集。。。具体的可以参考论文实验部分(Section 4)。

我最近会尝试上传一份到百度云盘哈。