search

Lornatang / SRGAN-PyTorch

A simple and complete implementation of super-resolution paper.
Apache License 2.0
410 stars 105 forks source link

Issues running the test.py #88

Open acamargofb opened 10 months ago

acamargofb commented 10 months ago

test.py --config_path ./configs/test/SRGAN_x4-SRGAN_ImageNet-Set5.yaml Traceback (most recent call last): File "test.py", line 26, in from dataset import CUDAPrefetcher, PairedImageDataset File "/mnt/c/Users/User/Desktop/Personal/UNI/SRGAN/SRGAN-PyTorch-main/SRGAN-PyTorch-main/dataset.py", line 25, in from imgproc import image_to_tensor, image_resize File "/mnt/c/Users/User/Desktop/Personal/UNI/SRGAN/SRGAN-PyTorch-main/SRGAN-PyTorch-main/imgproc.py", line 514, in center: tuple[(int, int)] = None, TypeError: 'type' object is not subscriptable

Lornatang commented 10 months ago

I use Python=3.10.x

acamargofb commented 10 months ago

Thanks, it is working now

acamargofb commented 10 months ago

One question, what is the maximum size of image that this model can handle as input? I mean 256x256x3? what value?. or this depends on the GPU memory?

Lornatang commented 10 months ago

Yeah. For GPU memory~