From 8a8e15e37a34ac72627d32a2ee8bba036f73a019 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 26 Jun 2018 17:15:26 +0200 Subject: dataset --- run.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index 80b3ae9..850c313 100644 --- a/run.py +++ b/run.py @@ -199,11 +199,11 @@ def store_frames(frames, outputPath, inputFirst=None, inputSecond=None, endpoint print('frames: {}'.format(len(frames))) writer = FFMPEG_VideoWriter(outputPath, (1024, 512), 25) if inputFirst is not None: - writer.write_frame(inputFirst) + writer.write_frame(tensor_to_image(inputFirst)) for frame in frames: writer.write_frame(tensor_to_image(frame)) if inputSecond is not None: - writer.write_frame(inputSecond) + writer.write_frame(tensor_to_image(inputSecond)) writer.close() if endpoint is not None: @@ -266,6 +266,7 @@ elif arguments_steps == 0: tensorInputSecond = load_image_tensor(arguments_strSecond) process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput) PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut) + elif arguments_mixImages: print("{} => {}".format(arguments_strFirst, arguments_strSecond)) inputFirst = load_image(os.path.join(arguments_strFirst, "frame_{:0>5}.png".format(int(arguments_aOffset)+1))) -- cgit v1.2.3-70-g09d2