diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 9 |
1 files changed, 8 insertions, 1 deletions
@@ -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() |