From dd006ec9e3a4dfd611f82deb399c3f5071a4e7b3 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 19 Apr 2018 02:27:58 +0200 Subject: test --- test.py | 50 +++++++++++++++++++++++++------------------------- 1 file changed, 25 insertions(+), 25 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index f40793a..2a60a65 100644 --- a/test.py +++ b/test.py @@ -24,31 +24,31 @@ 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() + # 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) -- cgit v1.2.3-70-g09d2