summaryrefslogtreecommitdiff
path: root/run.py
diff options
context:
space:
mode:
Diffstat (limited to 'run.py')
-rw-r--r--run.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/run.py b/run.py
index 62e822e..046473a 100644
--- a/run.py
+++ b/run.py
@@ -93,6 +93,7 @@ else:
process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput)
PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut)
else:
+ print("{} => {}".format(arguments_strFirst, arguments_strSecond))
inputFirst = PIL.Image.open(arguments_strFirst)
inputSecond = PIL.Image.open(arguments_strSecond)
tensorInputFirst = numpy.rollaxis(numpy.asarray(inputFirst)[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0