diff options
| -rw-r--r-- | process-sequences.sh | 2 | ||||
| -rw-r--r-- | run.py | 10 |
2 files changed, 7 insertions, 5 deletions
diff --git a/process-sequences.sh b/process-sequences.sh index d41dd9f..326da41 100644 --- a/process-sequences.sh +++ b/process-sequences.sh @@ -6,7 +6,7 @@ function process() { --mix-videos True \ --first "../pix2pixhd/sequences/$sequence_a/" \ --second "../pix2pixhd/sequences/$sequence_b/" \ - --steps 128 \ + --steps 32 \ --video-out "morph_video_${sequence_a}.mp4" scp renders/* jules@lmno:asdf/neural/morph/ } @@ -119,14 +119,16 @@ def process_tree(moduleNetwork, a, b, tensorOutput, steps): right = process_tree(moduleNetwork, middle, b, tensorOutput, steps / 2) return left + [np_middle] + right -def store_frames(frames, outputPath): +def store_frames(frames, outputPath, inputFirst=None, inputSecond=None): print('writing {}'.format(outputPath)) print('frames: {}'.format(len(frames))) writer = FFMPEG_VideoWriter(outputPath, (1024, 512), 25) - writer.write_frame(inputFirst) + if inputFirst is not None: + writer.write_frame(inputFirst) for frame in frames: writer.write_frame(frame) - writer.write_frame(inputSecond) + if inputSecond is not None: + writer.write_frame(inputSecond) def load_image(path): return numpy.rollaxis(numpy.asarray(PIL.Image.open(path))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 @@ -169,4 +171,4 @@ else: inputSecond = load_image(arguments_strSecond) frames = process_tree(moduleNetwork, inputFirst, inputSecond, tensorOutput, arguments_steps) outputPath = './renders/' + arguments_strVideoOut - store_frames(frames, outputPath) + store_frames(frames, outputPath, inputFirst, inputSecond) |