summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--run.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/run.py b/run.py
index a3bcc7c..55aa30f 100644
--- a/run.py
+++ b/run.py
@@ -92,7 +92,9 @@ else:
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)
+ outputPath = './renders/' + arguments_strVideoOut
+ print('writing {}'.format(outputPath))
+ writer = FFMPEG_VideoWriter(outputPath, (1024, 512), 25)
writer.write_frame(inputFirst)
for frame in tree:
writer.write_frame(frame)