expand

1 files changed, 3 insertions, 1 deletions

diff --git a/test.py b/test.py
index d41c8f2..7af6042 100644
--- a/test.py
+++ b/test.py
@@ -48,11 +48,13 @@ if __name__ == '__main__':
         if dataset.name() == 'RecursiveDatasetDataLoader':
             print(visuals.keys())
             im = visuals['fake_B']
-            save_path = render_dir + "frame_{:05d}.png".format(i+1)
+            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')
             image_pil = Image.fromarray(im)
             image_pil = ImageOps.expand(image_pil, 4)
             image_pil.save(save_path)
+            os.rename(save_path, final_path)
     webpage.save()
 
     os.remove(render_dir + "frame_00000.png")