From ce0db1254140e10cc8917babb497364e2aef7e8f Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Thu, 19 Dec 2019 18:39:03 +0100
Subject: type

---
 inversion/live.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'inversion/live.py')

diff --git a/inversion/live.py b/inversion/live.py
index dad916a..299aa3f 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -143,7 +143,7 @@ class Interpolator:
     gen_img = generator(gen_in, signature=gen_signature)
 
     # Convert generated image to channels_first.
-    self.gen_img = tf.transpose(gen_img, [0, 3, 1, 2])
+    # self.gen_img = tf.transpose(gen_img, [0, 3, 1, 2])
 
   def get_feed_dict(self):
     opt = {}
@@ -221,6 +221,7 @@ class Listener:
         print("Exiting...")
         break
       if (i % 100) == 0:
+        print(gen_images.shape)
         print("Generation time: {:.1f}s".format(time.time() - gen_time))
       out_img = vs.data2pil(gen_images[0])
       if out_img is not None:
-- 
cgit v1.2.3-70-g09d2