作者您好！代码问题想请教

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

作者您好！代码问题想请教 #59

Closed MrWan001 closed 8 months ago

MrWan001 commented 8 months ago

在DenoisingUNet和DenoisingNAFNet的代码中，我理解的输入是Xt和cond拼接，Xt是时刻t的状态，cond是低质图像作为条件，两个3通道拼接好之后是6通道，但是我看代码中是X=Xt-cond，然后X和cond拼接，不太明白Xt-cond是什么操作，我理解的Xt和cond有问题吗期待您的回复！

Algolzw commented 8 months ago

这是因为在公式中算score时有个 $x_t - \mu_t$项，其中\mu_t含有cond项，即score计算时有$x_t - cond$项。所以我在代码中使用$x_t - cond$代替x_t期望使预测更加容易。

MrWan001 commented 8 months ago

好的，明白了，谢谢