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
545 stars 41 forks source link

关于dehazing #39

Open bigwahaha opened 1 year ago

bigwahaha commented 1 year ago

您好，麻烦请教一下在Image Restoration with Mean-Reverting Stochastic Differential Equations附录中 image 在indoor数据集上的结果是采用哪个代码进行训练的呢？是否有预训练模型呀？

Algolzw commented 1 year ago

抱歉清理代码时这个dehazing的weight忘记保存了= =，不过你可以很容易修改deraining的代码（只需要替换数据集路径）来训练dehazing模型。训练大概2天左右。

bigwahaha commented 1 year ago

在indoor数据集hazy数据有十个雾霾等级，是选取其中一个等级进行训练的还是将十个等级一起进行训练的呀？

Algolzw commented 1 year ago

是一起训练的（不过只选取其中一个等级肯定会更好）。