1 files changed, 3 insertions, 2 deletions
diff --git a/test.py b/test.py
index 2f02e83..7d50ad6 100644
--- a/test.py
+++ b/test.py
@@ -49,8 +49,9 @@ if __name__ == '__main__':
             im = visuals['fake_B']
             save_path = render_dir + "frame_{:05d}.png".format(i+1)
             im = imresize(im, (248, 248), interp='bicubic')
-            im = ImageOps.expand(im, 4)
-            util.save_image(im, save_path)
+            image_pil = Image.fromarray(im)
+            image_pil = ImageOps.expand(image_pil, 4)
+            image_pil.save(image_path)
     webpage.save()
 
     os.remove(render_dir + "frame_00000.png")