1 files changed, 2 insertions, 1 deletions
diff --git a/test.py b/test.py
index 2c73c16..2f02e83 100644
--- a/test.py
+++ b/test.py
@@ -8,6 +8,7 @@ from util import html
 from shutil import move, copyfile
 from PIL import Image, ImageOps
 from skimage.transform import resize
+from scipy.misc import imresize
 
 import subprocess
 from time import sleep
@@ -47,7 +48,7 @@ if __name__ == '__main__':
         if dataset.name() == 'RecursiveDatasetDataLoader':
             im = visuals['fake_B']
             save_path = render_dir + "frame_{:05d}.png".format(i+1)
-            im = im.resize((248, 248))
+            im = imresize(im, (248, 248), interp='bicubic')
             im = ImageOps.expand(im, 4)
             util.save_image(im, save_path)
     webpage.save()