more general

1 files changed, 16 insertions, 8 deletions

diff --git a/test.py b/test.py
index 066c6c1..f5e9dd7 100644
--- a/test.py
+++ b/test.py
@@ -3,17 +3,24 @@ from options.test_options import TestOptions
 from data import CreateRecursiveDataLoader
 from models import create_model
 from util.visualizer import Visualizer
-from util import html
-
+from util import html, mkdir
+from shutil import movefile
 
 import subprocess
 
+
 if __name__ == '__main__':
     opt = TestOptions().parse()
     opt.nThreads = 1   # test code only supports nThreads = 1
     opt.batchSize = 1  # test code only supports batchSize = 1
     opt.serial_batches = True  # no shuffle
     opt.no_flip = True  # no flip
+    opt.experiment = opt.start_img.split(".")[0]
+
+    render_dir = self.opt.results_dir + "/" + self.opt.name + "/exp:" + opt.experiment
+
+    mkdir(render_dir)
+    movefile(opt.start_img, render_dir)
 
     data_loader = CreateRecursiveDataLoader(opt)
     dataset = data_loader.load_data()
@@ -35,11 +42,12 @@ 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 = "results/woodscaled_4_pix2pix/test_latest/images/" + ims[1]
-            next_image = "recursive/frame_{:04d}.png".format(i+1)
-            cmd = ("/bin/cp", last_image, next_image)
-            process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
-            output, error = process.communicate()
-            dataset.append(next_image)
+            last_image = opt.results_dir + "/" + opt.name + "/test_latest/images/" + ims[1]
+            next_image = opt.results_dir + "/" + opt.name + "/exp:" + opt.experiment + "/frame_{:05d}.png".format(i+1)
+            movefile(last_image, next_image)
 
     webpage.save()
+
+    cmd = ("/usr/bin/ffmpeg", "-i", "frame_%04d.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", opt.name + "_" + opt.experiment + ".mp4")
+    process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
+    output, error = process.communicate()