From 97272e6bde3ebf1a409489aeecd11bc98a3bb193 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 15:38:38 +0200 Subject: go hey --- run.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index 8501d38..a3bcc7c 100644 --- a/run.py +++ b/run.py @@ -87,11 +87,13 @@ else: 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) else: - tensorInputFirst = numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strFirst))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 - tensorInputSecond = numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strSecond))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 + inputFirst = PIL.Image.open(arguments_strFirst) + inputSecond = PIL.Image.open(arguments_strSecond) + tensorInputFirst = numpy.rollaxis(numpy.asarray(inputFirst)[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 + tensorInputSecond = numpy.rollaxis(numpy.asarray(inputSecond)[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 tree = process_tree(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput, arguments_steps) writer = FFMPEG_VideoWriter('./renders/' + arguments_strVideoOut, (1024, 512), 25) - writer.write_frame(tensorInputFirst) + writer.write_frame(inputFirst) for frame in tree: writer.write_frame(frame) - writer.write_frame(tensorInputSecond) + writer.write_frame(inputSecond) -- cgit v1.2.3-70-g09d2