summaryrefslogtreecommitdiff
path: root/recursive.py
diff options
context:
space:
mode:
authorjules@lens <julescarbon@gmail.com>2018-09-05 12:59:08 +0200
committerjules@lens <julescarbon@gmail.com>2018-09-05 12:59:08 +0200
commitc2421b5c8eea3b3b3769ac64119e4445e616e963 (patch)
tree3b997b6450c3202727382d1e12e41d4eae827341 /recursive.py
parent3e66eb779b23be9f4cfdc3fc57c523ea2a0fd5c4 (diff)
augment script
Diffstat (limited to 'recursive.py')
-rw-r--r--recursive.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/recursive.py b/recursive.py
index 5ab6ff7..bacc3ce 100644
--- a/recursive.py
+++ b/recursive.py
@@ -62,6 +62,8 @@ for i, data in enumerate(dataset):
data['label'] = data['label'].uint8()
data['inst'] = data['inst'].uint8()
minibatch = 1
+ print(data['label'])
+ print(data['inst'])
generated = model.inference(data['label'], data['inst'])
last_path = opt.render_dir + "frame_{:05d}.png".format(i)