aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-12-03 20:40:11 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-12-03 20:40:29 +0300
commitcefb5d6d7dbb35e68467bb7965f7139abfaf290d (patch)
tree4392902a7ff807d67ae388f0fba3332e547a1b7c /extensions-builtin
parent8fba733c0906dd3a80c0a3873793cffa4c78ce04 (diff)
fix accessing options when they are not ready for SwinIR.
Diffstat (limited to 'extensions-builtin')
-rw-r--r--extensions-builtin/SwinIR/scripts/swinir_model.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py
index 782769e2..9a74b253 100644
--- a/extensions-builtin/SwinIR/scripts/swinir_model.py
+++ b/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