summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-26 17:09:44 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-26 17:09:44 +0200
commitd626b4b48e02649bf4cbf91864b580a5d4edae51 (patch)
treeb53ced8ba917d7f95307c1eb1e75a5d0b7b2cc27
parent9a41306f1e27f40bd13d941e56559ca2240fcda2 (diff)
dataset
-rw-r--r--run.py4
1 files changed, 2 insertions, 2 deletions
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)