RaoUmer/SRResCGAN

how to train for gray image?

Opened this issue · 1 comments

Thanks for sharing! the trainning code just for color image ,are there any instructions to use your code for training gray image?

Hi, For the gray-scale, in the main_sr_color.py at code-line #L767, pass the color=False for the generator network settings, and for discriminator network setting, set the --in_nc as 1 at the #L43. Besides that, prepare the training and validation data according to gray-scale images.