diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 18:39:03 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 18:39:03 +0100 |
| commit | ce0db1254140e10cc8917babb497364e2aef7e8f (patch) | |
| tree | f4bc732cf8e5a48207c5a7d786ae628eb2e57dfb /inversion/live.py | |
| parent | 2207d789c30d03e585028f0af80d0bc22623cbde (diff) | |
type
Diffstat (limited to 'inversion/live.py')
| -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: |