diff options
| -rw-r--r-- | test.py | 7 |
1 files changed, 4 insertions, 3 deletions
@@ -9,6 +9,7 @@ from shutil import move, copyfile from PIL import Image, ImageOps from skimage.transform import resize from scipy.misc import imresize +from shutil import copyfile import subprocess from time import sleep @@ -53,16 +54,16 @@ if __name__ == '__main__': s = 256 p = 8 im = imresize(im, (s-p, s-p), interp='bicubic') + image_pil.save(save_path) image_pil = Image.fromarray(im) image_pil = ImageOps.expand(image_pil, p) image_pil.save(save_path) - os.rename(save_path, final_path) + copyfile(save_path, final_path) webpage.save() os.remove(render_dir + "frame_00000.png") - - cmd = ("/usr/bin/ffmpeg", "-i", opt.results_dir + opt.name + "/frame_%05d_fake_B.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_tmp.png", "-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() |