JingyunLiang / SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)

Home Page:https://arxiv.org/abs/2108.10257

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Replacement model problem

labels168 opened this issue · comments

I use the following model and it executes fine
python main_test_swinir.py --task real_sr --scale 4 --model_path model_zoo/swinir/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq testsets/RealSRSet+5images --tile

But if I change the model, there will be an error, I don't know how to fix this problem, please help, thanks!
python main_test_swinir.py --task real_sr --scale 4 --model_path model_zoo/swinir/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth --folder_lq testsets/RealSRSet+5images --tile

The difference is in the model:
003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth

I think 003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x2_GAN.pth is for scale 2, you can try change this parameter --scale 2