Update inference_coz.py

inference_coz.py

@@ -295,7 +295,11 @@ if __name__ == "__main__":
                 prev_sr_output_pil = Image.open(prev_sr_output_path).convert('RGB')
                 rscale = args.upscale
                 w, h = prev_sr_output_pil.size
-                new_w, new_h = w // rscale, h // rscale
+
+                if rscale != 0:
+                    new_w, new_h = w // rscale, h // rscale
+                else:
+                    new_w, new_h = w, h
                 cropped_region = prev_sr_output_pil.crop(((w-new_w)//2, (h-new_h)//2, (w+new_w)//2, (h+new_h)//2))
                 current_sr_input_image_pil = cropped_region.resize((w, h), Image.BICUBIC)
 
@@ -307,7 +311,10 @@ if __name__ == "__main__":
             elif args.rec_type == 'recursive_multiscale':
                 prev_sr_output_path = f'{rec_dir}/{rec}.png'
                 prev_sr_output_pil = Image.open(prev_sr_output_path).convert('RGB')
-                rscale = args.upscale
+                if rscale != 0:
+                    new_w, new_h = w // rscale, h // rscale
+                else:
+                    new_w, new_h = w, h
                 w, h = prev_sr_output_pil.size
                 new_w, new_h = w // rscale, h // rscale
                 cropped_region = prev_sr_output_pil.crop(((w-new_w)//2, (h-new_h)//2, (w+new_w)//2, (h+new_h)//2))