Hard-coded number of output channels?

[RunDevelopment]

Hi! I recently looked at the code for DITN and noticed that the number of output channels seems to be hard-coded to 3 (RGB):

DITN/models/DITN_Real.py

Line 220 in 3438e42

self.upsample = UpsampleOneStep(upscale, dim, 3)

Did I read that correctly?

If yes, do you think it would be possible to change the number of output channels from 3 to inp_channels? This would allow us to train grayscale and even RGBA models.

[yongliuy]

Hi~

(1) What you read is correct.
(2) I think changing the number of output channels from 3 to other is ok, but the model may need to be retrained.