ShawkhIbneRashid / medical-images-sr

Medical image super-resolution using Real-ESRGAN and ESRGAN.
30 stars 3 forks source link

final testing have channel number issue #6

Open TPIOS opened 1 year ago

TPIOS commented 1 year ago

Hi, I am getting errors in testing realESRGAN after smoothly running the training.

The line that encounters the error is

output = upsampler.enhance(img, outscale=outscale)

, where the img shape I check is (64,64).

The error it gives:

RuntimeError: Given groups=1, weight of size [64, 1, 3, 3], expected input[1, 3, 64, 64] to have 1 channels, but got 3 channels instead

Any idea about it?

Thank you very much!