diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:38:38 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:38:38 +0200 |
| commit | 97272e6bde3ebf1a409489aeecd11bc98a3bb193 (patch) | |
| tree | de349f1750911bb1d20f07237f86e265f97e9b35 /run.py | |
| parent | 2d956623a8535472fdcc494cc33955d99918869d (diff) | |
go hey
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 10 |
1 files changed, 6 insertions, 4 deletions
@@ -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) |