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

网络输入Xt问题 #82

Open Edward-1997 opened 5 months ago

Edward-1997 commented 5 months ago

作者,你好,我观察网络的输入不是Xt是,是Xt-cond然后和cond做concat,这是为什么 image 这种输入比直接输入Xt效果要好吗

Algolzw commented 5 months ago

你好,我这样处理是根据论文中 公式6和8算score时发现x_t - m_t刚好包含了xt - cond这一项,因此就直接在网络里这样操作了。但具体是否会有提升还没有测试过,理论上效果应该差不多。