Explorar o código

fix accessing options when they are not ready for SwinIR.

AUTOMATIC %!s(int64=2) %!d(string=hai) anos
pai
achega
cefb5d6d7d
Modificáronse 1 ficheiros con 6 adicións e 2 borrados
  1. 6 2
      extensions-builtin/SwinIR/scripts/swinir_model.py

+ 6 - 2
extensions-builtin/SwinIR/scripts/swinir_model.py

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