fix accessing options when they are not ready for SwinIR.
This commit is contained in:
parent
8fba733c09
commit
cefb5d6d7d
1 changed files with 6 additions and 2 deletions
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue