y

2 files changed, 8 insertions, 5 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index e42f1f7..32b0018 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -21,7 +21,6 @@ class RecursiveDataset(BaseDataset):
 
     def __getitem__(self, index):
         A_path = self.opt.results_dir + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index)
-        print('next image: ' + A_path)
         while not os.path.exists(A_path):
             time.sleep(0.1)
         A_img = Image.open(A_path).convert('RGB')
diff --git a/test.py b/test.py
index 39fe4c5..cd6fb7a 100644
--- a/test.py
+++ b/test.py
@@ -19,10 +19,10 @@ if __name__ == '__main__':
     opt.no_flip = True  # no flip
     opt.experiment = opt.start_img.split(".")[0]
 
-    render_dir = opt.results_dir + "/" + opt.name + "/exp:" + opt.experiment
+    render_dir = opt.results_dir + "/" + opt.name + "/exp:" + opt.experiment + "/"
 
     mkdirs(render_dir)
-    copyfile(opt.start_img, render_dir + "/frame_00000.png")
+    copyfile(opt.start_img, render_dir + "frame_00000.png")
 
     data_loader = CreateRecursiveDataLoader(opt)
     dataset = data_loader.load_data()
@@ -45,11 +45,15 @@ if __name__ == '__main__':
         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 = opt.results_dir + opt.name + "/exp:" + opt.experiment + "/frame_{:05d}.png".format(i+1)
+            next_image = render_dir + "frame_{:05d}.png".format(i+1)
             move(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")
+    cmd = ("/usr/bin/ffmpeg", "-i", "frame_%04d.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()
+
+    cmd = ("scp", render_dir + opt.name + "_" + opt.experiment + ".mp4", "jules@asdf.us:asdf/neural/")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()