How can I apply 512*512 images on ESRGAN?

Open Diagreen opened this issue 3 years ago • 0 comments

Hi, I want train ESRGAN with 512*512 RGB images. When I run the training ESRGAN after PSNR one, I met AssertionError on discriminator_arch.py. On this code, It can input only 128 or 256 maybe. How can I use my data set without downsampling?

Thank you.

Jun 29 '22 14:06 Diagreen