y

1 files changed, 4 insertions, 4 deletions

diff --git a/test.py b/test.py
index 0733b7b..a13ee0f 100644
--- a/test.py
+++ b/test.py
@@ -5,7 +5,7 @@ from models import create_model
 from util.visualizer import Visualizer
 from util.util import mkdirs
 from util import html
-from shutil import move
+from shutil import move, copyfile
 
 import subprocess
 from time import sleep
@@ -22,7 +22,7 @@ if __name__ == '__main__':
     render_dir = opt.results_dir + "/" + opt.name + "/exp:" + opt.experiment
 
     mkdirs(render_dir)
-    move(opt.start_img, render_dir)
+    copyfile(opt.start_img, render_dir)
 
     data_loader = CreateRecursiveDataLoader(opt)
     dataset = data_loader.load_data()
@@ -44,8 +44,8 @@ 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 = 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)
+            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)
             move(last_image, next_image)
 
     webpage.save()