From c2421b5c8eea3b3b3769ac64119e4445e616e963 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Wed, 5 Sep 2018 12:59:08 +0200 Subject: augment script --- recursive.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'recursive.py') 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) -- cgit v1.2.3-70-g09d2