Update swinir.py
This commit is contained in:
parent
403bb02948
commit
a5313351b3
|
@ -55,8 +55,8 @@ def load_models(dirname):
|
||||||
def upscale(
|
def upscale(
|
||||||
img,
|
img,
|
||||||
model,
|
model,
|
||||||
tile=opts.GAN_tile,
|
tile=opts.SWIN_tile,
|
||||||
tile_overlap=opts.GAN_tile_overlap,
|
tile_overlap=opts.SWIN_tile_overlap,
|
||||||
window_size=8,
|
window_size=8,
|
||||||
scale=4,
|
scale=4,
|
||||||
):
|
):
|
||||||
|
|
Loading…
Reference in New Issue