From 468af07857297ba0cbf01eda2442501dc7351c33 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 19 Dec 2019 18:21:57 +0100 Subject: version --- inversion/live.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'inversion') diff --git a/inversion/live.py b/inversion/live.py index afeeccb..7f07189 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -129,14 +129,17 @@ class Interpolator: } def build(self): - lerp_z = lerp(self.opts, 'latent', [BATCH_SIZE, Z_DIM]) - sin_z = sin(self.opts, 'orbit', [BATCH_SIZE, Z_DIM]) - lerp_label = lerp(self.opts, 'label', [BATCH_SIZE, N_CLASS]) + # lerp_z = lerp(self.opts, 'latent', [BATCH_SIZE, Z_DIM]) + # sin_z = sin(self.opts, 'orbit', [BATCH_SIZE, Z_DIM]) + # lerp_label = lerp(self.opts, 'label', [BATCH_SIZE, N_CLASS]) + + self.opts['z'] = InterpolatorParam(name='z', shape=[BATCH_SIZE, Z_DIM]) + self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS]) gen_in = {} gen_in['truncation'] = 1.0 # self.opts['truncation'].variable - gen_in['z'] = lerp_z # + sin_z - gen_in['y'] = lerp_label + gen_in['z'] = self.opts['z'].variable # lerp_z # + sin_z + gen_in['y'] = self.opts['y'].variable gen_img = generator(gen_in, signature=gen_signature) # Convert generated image to channels_first. -- cgit v1.2.3-70-g09d2