From f3dac7ce89a872eb40aca3fae7866c224cd494b3 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 20 Apr 2018 11:56:56 +0200 Subject: try resizing --- test.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 122f563..c77298f 100644 --- a/test.py +++ b/test.py @@ -6,11 +6,12 @@ from util.visualizer import Visualizer from util.util import mkdirs from util import html from shutil import move, copyfile +from scipy.misc import imresize +from PIL import Image, ImageOps import subprocess from time import sleep - if __name__ == '__main__': opt = TestOptions().parse() opt.nThreads = 1 # test code only supports nThreads = 1 @@ -44,14 +45,15 @@ if __name__ == '__main__': print('%04d: process image... %s' % (i, img_path)) ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) if dataset.name() == 'RecursiveDatasetDataLoader': - last_image = opt.results_dir + opt.name + "/test_latest/images/" + ims[1] - next_image = render_dir + "frame_{:05d}.png".format(i+1) - move(last_image, next_image) - + im = visuals['fake_B'] + save_path = render_dir + "frame_{:05d}.png".format(i+1) + im = imresize(im, 248, 248, interp='bicubic') + im = ImageOps.expand(im, (4,4,4,4)) + util.save_image(im, save_path) webpage.save() 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") process = subprocess.Popen(cmd, stdout=subprocess.PIPE) output, error = process.communicate() -- cgit v1.2.3-70-g09d2