expand

1 files changed, 4 insertions, 2 deletions

diff --git a/test.py b/test.py
index 7af6042..6bda757 100644
--- a/test.py
+++ b/test.py
@@ -50,9 +50,11 @@ if __name__ == '__main__':
             im = visuals['fake_B']
             save_path = render_dir + "frame_{:05d}_tmp.png".format(i+1)
             final_path = render_dir + "frame_{:05d}.png".format(i+1)
-            im = imresize(im, (248, 248), interp='bicubic')
+            s = 256
+            p = 8
+            im = imresize(im, (s-p, s-p), interp='bicubic')
             image_pil = Image.fromarray(im)
-            image_pil = ImageOps.expand(image_pil, 4)
+            image_pil = ImageOps.expand(image_pil, p)
             image_pil.save(save_path)
             os.rename(save_path, final_path)
     webpage.save()