diff options
| -rw-r--r-- | test.py | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -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() |