iSEE-Laboratory/DiffUIR

support grayscale image training

Closed this issue · 1 comments

xxk123456 commented

Great Works!! Does the model support grayscale image training? What parameters need to be changed?

zhengdian1 commented

Of course. You can change the Line 513 in ./src/model.py (change the channel from 3 to 1).