test

1 files changed, 25 insertions, 25 deletions

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)
+    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()
+    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")
+    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()