From 4d934271a8d3af37cb39c1e9e8c1ca125ed58663 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 19 Apr 2018 02:31:06 +0200 Subject: test --- test.py | 54 +++++++++++++++++++++++++++--------------------------- 1 file changed, 27 insertions(+), 27 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 34fbc1f..65f334f 100644 --- a/test.py +++ b/test.py @@ -24,33 +24,33 @@ if __name__ == '__main__': mkdirs(render_dir) copyfile(opt.start_img, render_dir + "frame_00000.png") - # data_loader = CreateRecursiveDataLoader(opt) - # dataset = data_loader.load_data() - # ds = dataset.dataset - # model = create_model(opt) - # visualizer = Visualizer(opt) - # # create website - # web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) - # webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) - # # test - # print(dataset.name()) - # for i, data in enumerate(data_loader): - # if i >= opt.how_many: - # break - # model.set_input(data) - # model.test() - # visuals = model.get_current_visuals() - # img_path = model.get_image_paths() - # 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) - - # webpage.save() - - 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") + data_loader = CreateRecursiveDataLoader(opt) + dataset = data_loader.load_data() + ds = dataset.dataset + model = create_model(opt) + visualizer = Visualizer(opt) + # create website + web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch)) + webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch)) + # test + print(dataset.name()) + for i, data in enumerate(data_loader): + if i >= opt.how_many: + break + model.set_input(data) + model.test() + visuals = model.get_current_visuals() + img_path = model.get_image_paths() + 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) + + webpage.save() + + 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