From 2dfafe809a5c800ae11b22eb4bc7de0ac3e549a3 Mon Sep 17 00:00:00 2001 From: sniklaus Date: Fri, 2 Feb 2018 18:07:32 -0800 Subject: no message --- run.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/run.py b/run.py index a7a7add..0020c08 100644 --- a/run.py +++ b/run.py @@ -170,8 +170,8 @@ moduleNetwork = Network().cuda() ########################################################## -tensorInputFirst = torch.FloatTensor(numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strFirst))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0) -tensorInputSecond = torch.FloatTensor(numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strSecond))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0) +tensorInputFirst = torch.FloatTensor(numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strFirst))[:, :, ::-1], 2, 0).astype(numpy.float32) / 255.0) +tensorInputSecond = torch.FloatTensor(numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strSecond))[:, :, ::-1], 2, 0).astype(numpy.float32) / 255.0) tensorOutput = torch.FloatTensor() assert(tensorInputFirst.size(1) == tensorInputSecond.size(1)) @@ -232,4 +232,4 @@ if True: tensorOutput = tensorOutput.cpu() # end -PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut) \ No newline at end of file +PIL.Image.fromarray((numpy.rollaxis(tensorOutput.clamp(0.0, 1.0).numpy(), 0, 3)[:, :, ::-1] * 255.0).astype(numpy.uint8)).save(arguments_strOut) \ No newline at end of file -- cgit v1.2.3-70-g09d2