From 15f2f4ba78e07dffd1f7454eb43c6660868dbf4b Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 15:35:08 +0200 Subject: go hey --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index c705552..77be351 100644 --- a/run.py +++ b/run.py @@ -54,7 +54,7 @@ def process_tree(moduleNetwork, a, b, tensorOutput, steps): tensorInputSecond = torch.FloatTensor(b) process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput) middle = tensorOutput.clamp(0.0, 1.0).numpy() - np_middle = (numpy.rollaxis(output, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8) + np_middle = (numpy.rollaxis(middle, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8) if steps < 2: return [np_middle] else: -- cgit v1.2.3-70-g09d2