From d626b4b48e02649bf4cbf91864b580a5d4edae51 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 26 Jun 2018 17:09:44 +0200 Subject: dataset --- run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index 33f2c7d..80b3ae9 100644 --- a/run.py +++ b/run.py @@ -268,8 +268,8 @@ elif arguments_steps == 0: PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut) elif arguments_mixImages: print("{} => {}".format(arguments_strFirst, arguments_strSecond)) - inputFirst = load_image(os.path.join(arguments_strFirst, "frame_{:0>5}.png".format(int(arguments_aOffset)))) - inputSecond = load_image(os.path.join(arguments_strSecond, "frame_{:0>5}.png".format(int(arguments_bOffset)))) + inputFirst = load_image(os.path.join(arguments_strFirst, "frame_{:0>5}.png".format(int(arguments_aOffset)+1))) + inputSecond = load_image(os.path.join(arguments_strSecond, "frame_{:0>5}.png".format(int(arguments_bOffset)+1))) outputPath = './renders/' + arguments_strVideoOut frames = process_tree(moduleNetwork, inputFirst, inputSecond, tensorOutput, arguments_steps * arguments_dilate, arguments_dilate) frames = smooth_frames(moduleNetwork, tensorOutput, frames, arguments_smooth) -- cgit v1.2.3-70-g09d2