Spaces:
Running
on
Zero
Running
on
Zero
Update inference_coz.py
Browse files- inference_coz.py +3 -2
inference_coz.py
CHANGED
@@ -311,12 +311,13 @@ if __name__ == "__main__":
|
|
311 |
elif args.rec_type == 'recursive_multiscale':
|
312 |
prev_sr_output_path = f'{rec_dir}/{rec}.png'
|
313 |
prev_sr_output_pil = Image.open(prev_sr_output_path).convert('RGB')
|
|
|
|
|
|
|
314 |
if rscale != 0:
|
315 |
new_w, new_h = w // rscale, h // rscale
|
316 |
else:
|
317 |
new_w, new_h = w, h
|
318 |
-
w, h = prev_sr_output_pil.size
|
319 |
-
new_w, new_h = w // rscale, h // rscale
|
320 |
cropped_region = prev_sr_output_pil.crop(((w-new_w)//2, (h-new_h)//2, (w+new_w)//2, (h+new_h)//2))
|
321 |
current_sr_input_image_pil = cropped_region.resize((w, h), Image.BICUBIC)
|
322 |
|
|
|
311 |
elif args.rec_type == 'recursive_multiscale':
|
312 |
prev_sr_output_path = f'{rec_dir}/{rec}.png'
|
313 |
prev_sr_output_pil = Image.open(prev_sr_output_path).convert('RGB')
|
314 |
+
rscale = args.upscale
|
315 |
+
w, h = prev_sr_output_pil.size
|
316 |
+
|
317 |
if rscale != 0:
|
318 |
new_w, new_h = w // rscale, h // rscale
|
319 |
else:
|
320 |
new_w, new_h = w, h
|
|
|
|
|
321 |
cropped_region = prev_sr_output_pil.crop(((w-new_w)//2, (h-new_h)//2, (w+new_w)//2, (h+new_h)//2))
|
322 |
current_sr_input_image_pil = cropped_region.resize((w, h), Image.BICUBIC)
|
323 |
|