Commit cefb5d6d authored by AUTOMATIC's avatar AUTOMATIC

fix accessing options when they are not ready for SwinIR.

parent 8fba733c
......@@ -99,11 +99,15 @@ class UpscalerSwinIR(Upscaler):
def upscale(
img,
model,
tile=opts.SWIN_tile,
tile_overlap=opts.SWIN_tile_overlap,
tile=None,
tile_overlap=None,
window_size=8,
scale=4,
):
tile = tile or opts.SWIN_tile
tile_overlap = tile_overlap or opts.SWIN_tile_overlap
img = np.array(img)
img = img[:, :, ::-1]
img = np.moveaxis(img, 2, 0) / 255
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment