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

我只想用于灰度图的噪音去除，训练和测试结果都为灰度图，请问如何修改参数

fsqvictor opened this issue 10 months ago · comments

fsqvictor commented 10 months ago

我只想用于灰度图的噪音去除，训练和测试结果都为灰度图，请问如何修改参数

balabala commented 10 months ago

去噪的话使用denoising-sde文件夹，一般来说只需要修改配置文件中datasets部分(color: gray)以及将模型部分输入输出通道改为1.

fsqvictor commented 10 months ago

好的，多谢多谢