From 7d3d1cb632a4d534149542165ec5e78cd46016fd Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 15:28:25 +0200 Subject: split out network, do a morph dewd --- run.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/run.py b/run.py index bbbeafc..744cb9f 100644 --- a/run.py +++ b/run.py @@ -1,5 +1,6 @@ #!/usr/bin/env python2.7 +import os import sys import getopt import numpy @@ -38,6 +39,8 @@ for strOption, strArgument in getopt.getopt(sys.argv[1:], '', [ strParameter[2:] arguments_steps = int(strArgument) # end # end +if not os.path.exists('./renders'): + os.mkdir('renders') moduleNetwork = Network(arguments_strModel).cuda() tensorOutput = torch.FloatTensor() @@ -76,7 +79,7 @@ else: PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut) else: tree = process_tree(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput, arguments_steps) - writer = FFMPEG_VideoWriter(arguments_strVideoOut, (1024, 512), 25) + writer = FFMPEG_VideoWriter('./renders/' + arguments_strVideoOut, (1024, 512), 25) writer.write_frame(tensorInputFirst) for frame in tree: writer.write_frame(frame) -- cgit v1.2.3-70-g09d2