Răsfoiți Sursa

fix accessing options when they are not ready for SwinIR.

AUTOMATIC 2 ani în urmă
părinte
comite
cefb5d6d7d
1 a modificat fișierele cu 6 adăugiri și 2 ștergeri
  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