diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:35:08 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:35:08 +0200 |
| commit | 15f2f4ba78e07dffd1f7454eb43c6660868dbf4b (patch) | |
| tree | dfe7368ba0f0ce8db24a1c26ada417a2761e0888 | |
| parent | edc10af27686f40778641e4e5b35df13d4cf8fad (diff) | |
go hey
| -rw-r--r-- | run.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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: |