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
521 stars 39 forks source link

灰度图训练，如何修改参数 #71

Open fsqvictor opened 6 months ago

fsqvictor commented 6 months ago

如果只是用于灰度图的训练，如何允许输入为灰度图，输出结果也为灰度图

Algolzw commented 6 months ago

可以把网络输入输出通道数改成1并把color改成gray.

fsqvictor commented 6 months ago

我试试