summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-26 17:08:56 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-26 17:08:56 +0200
commit9a41306f1e27f40bd13d941e56559ca2240fcda2 (patch)
treeec3cec51a146ad5bb5f722ae4adbf9248e05d62d
parent1cf17526b0654e1ecbf4f362c5361a7b52cbbb19 (diff)
dataset
-rw-r--r--run.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/run.py b/run.py
index c58afa2..33f2c7d 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(a_offset))))
- inputSecond = load_image(os.path.join(arguments_strSecond, "frame_{:0>5}.png".format(int(b_offset))))
+ 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))))
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)