diff options
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 |
