diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 24 |
1 files changed, 16 insertions, 8 deletions
@@ -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() |