try transitioni

1 files changed, 8 insertions, 1 deletions

diff --git a/test.py b/test.py
index 630e446..e43d293 100644
--- a/test.py
+++ b/test.py
@@ -10,6 +10,7 @@ from PIL import Image, ImageOps
 from skimage.transform import resize
 from scipy.misc import imresize
 from shutil import copyfile
+import numpy as np
 
 import subprocess
 from time import sleep
@@ -37,6 +38,7 @@ if __name__ == '__main__':
     webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch))
     # test
     print(dataset.name())
+    last_im = None
     for i, data in enumerate(data_loader):
         if i >= opt.how_many:
             break
@@ -58,6 +60,11 @@ if __name__ == '__main__':
             # image_pil = ImageOps.expand(image_pil, p)
             # image_pil.save(save_path)
             # copyfile(save_path, final_path)
+            if i < 1:
+                im /= 2
+                tmp_im = np.copy(im/2)
+                im += last_im
+                last_im = tmp_im
             image_pil = Image.fromarray(im)
             image_pil.save(save_path)
             os.rename(save_path, final_path)
@@ -65,7 +72,7 @@ if __name__ == '__main__':
 
     os.remove(render_dir + "frame_00000.png")
 
-    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "/frame_%05d.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
+    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "/frame_%05d.png", "-y", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()