fix accessing options when they are not ready for SwinIR.

This commit is contained in:
AUTOMATIC 2022-12-03 20:40:11 +03:00
parent 8fba733c09
commit cefb5d6d7d
1 changed files with 6 additions and 2 deletions

View File

@ -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