k

1 files changed, 8 insertions, 2 deletions

diff --git a/test.py b/test.py
index ad42b2d..b2c679c 100644
--- a/test.py
+++ b/test.py
@@ -6,6 +6,8 @@ from util.visualizer import Visualizer
 from util import html
 
 
+import subprocess
+
 if __name__ == '__main__':
     opt = TestOptions().parse()
     opt.nThreads = 1   # test code only supports nThreads = 1
@@ -33,7 +35,11 @@ if __name__ == '__main__':
         ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
         print(dataset.name())
         if dataset.name() == 'RecursiveDatasetDataLoader':
-            # dataset.append(ims)
-            print(ims)
+            last_image = "results/woodscaled_4_pix2pix/test_latest/images/" + ims[1]
+            next_image = "recursive/frame{:04d}".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)
 
     webpage.save()