SRResCGAN icon indicating copy to clipboard operation
SRResCGAN copied to clipboard

how to train for gray image?

Open aalala opened this issue 3 years ago • 1 comments

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

aalala avatar Mar 29 '21 15:03 aalala

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.

RaoUmer avatar Mar 30 '21 12:03 RaoUmer