diff options
Diffstat (limited to 'inversion')
| -rw-r--r-- | inversion/live.py | 3 |
1 files changed, 2 insertions, 1 deletions
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: |