Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

Home Page:https://algolzw.github.io/ir-sde/index.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

灰度图训练,如何修改参数

fsqvictor opened this issue · comments

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

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

我试试