From 9382d73eaab427d4de7fab926fd4762a44d19cf2 Mon Sep 17 00:00:00 2001 From: Nate Date: Thu, 21 May 2020 14:21:31 -0500 Subject: [PATCH] small fixes --- detector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/detector.py b/detector.py index 59bc73c..e048d36 100644 --- a/detector.py +++ b/detector.py @@ -216,7 +216,7 @@ def resize_GAN(self, img_path, img_name, is_video=False): return # Now run ESRGAN inference gan_img_path = self.out_path + img_name[:-4] + '.png' - self.esrgan_instance.run_esrgan(test_img_folder=file_name, out_filename=gan_img_path) + self.esrgan_instance.run_esrgan(test_img_folder=file_name, out_filename=gan_img_path, mosaic_res=granularity) else: try: video_path = img_path @@ -252,7 +252,7 @@ def resize_GAN(self, img_path, img_name, is_video=False): # run ESRGAN algorithms gan_img_path = self.out_path + img_name[:-4] + str(count).zfill(6) + '.png' - self.esrgan_instance.run_esrgan(test_img_folder=file_name, out_filename=gan_img_path) + self.esrgan_instance.run_esrgan(test_img_folder=file_name, out_filename=gan_img_path, mosaic_res=granularity) gan_image = skimage.io.imread(gan_img_path) gan_image = resize(gan_image, (image.shape[1], image.shape[0]))