From fe6ee784b746637b10ee000d1f63e47dbdf3fffb Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Fri, 20 Apr 2018 12:20:45 +0200
Subject: expand

---
 test.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'test.py')

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")
-- 
cgit v1.2.3-70-g09d2