diff options
Diffstat (limited to 'inversion')
| -rw-r--r-- | inversion/live.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/inversion/live.py b/inversion/live.py index 07af3fe..5bd26fe 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -129,17 +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], type='noise') - self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS], type='label') + # self.opts['z'] = InterpolatorParam(name='z', shape=[BATCH_SIZE, Z_DIM], type='noise') + # self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS], type='label') gen_in = {} gen_in['truncation'] = 1.0 # self.opts['truncation'].variable - gen_in['z'] = self.opts['z'].variable # lerp_z # + sin_z - gen_in['y'] = self.opts['y'].variable + gen_in['z'] = lerp_z + sin_z + gen_in['y'] = lerp_label self.gen_img = generator(gen_in, signature=gen_signature) # Convert generated image to channels_first. @@ -169,8 +169,8 @@ class Interpolator: sys.stderr.write('{} not a valid option\n'.format(key)) def on_step(self, i, sess): - self.opts['z'].randomize(truncation=1.0) - self.opts['y'].randomize(num_classes=1.0) + # self.opts['z'].randomize(truncation=1.0) + # self.opts['y'].randomize(num_classes=1.0) gen_images = sess.run(self.gen_img, feed_dict=self.get_feed_dict()) return gen_images |