alterzero/RBPN-PyTorch

could you share the code to train luminance channel,not RGB channel?

gittigxuy opened this issue · 4 comments

Thanks

You can do it.
Basically, you can change num_channels=1

No,I do not think so,just change num_channels=1,so could you please release a detail example code to train luminance channel data?Thanks

Of course you need to convert the train image to luminance only beforehand.

Anyhow, I did not train the network using only luminance channel, but only the evaluation is calculated the PSNR using luminance layer only.

So,if I train RGB data,and test with luminance channel,does the result same with both Train and Test use luminance channel