diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:48:23 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:48:23 +0200 |
| commit | c48c5c5ee5cf18b741be964777e48334d58fc0bb (patch) | |
| tree | 9099bc63d7eba8769397ef867355650435b57af4 /run.py | |
| parent | 26a08621599ed6a30137c41b32a4b8efac5ff97e (diff) | |
process script
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -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 |