When I try to upsample arbitrary size image (e.g., 228x344 Set5 woman.png), dimension mismatching error occurs at line 400 in unet.py file.
The problem happens in the else clause at line 404 in the same file, where the argument h is set to the width of feats (e.g., h=feats[-1].shape[-1]).
After I change it to h=feats[-1].shape[-2], arbitrary size image upsampling becomes possible.
If the code is not intended, then it would be nice to fix it.
When I try to upsample arbitrary size image (e.g., 228x344 Set5 woman.png), dimension mismatching error occurs at line 400 in unet.py file. The problem happens in the else clause at line 404 in the same file, where the argument h is set to the width of feats (e.g., h=feats[-1].shape[-1]). After I change it to h=feats[-1].shape[-2], arbitrary size image upsampling becomes possible. If the code is not intended, then it would be nice to fix it.
https://github.com/Ree1s/IDM/blob/d224c0ef885168ebd1d19ab4cf4a66a1939ffef0/model/sr3_modules/unet.py#L402