diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 14:00:32 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 14:00:32 +0100 |
| commit | 4d236ffc49f377d54d6a8a93e864c52f925c6ee9 (patch) | |
| tree | 6d336a32c23c5b4da2af1ffb6c2dc2f0d51b832f /inversion/live.py | |
| parent | ca032925551726fbca9e3fffa76aa766fdc37499 (diff) | |
clip latents differently
Diffstat (limited to 'inversion/live.py')
| -rw-r--r-- | inversion/live.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/inversion/live.py b/inversion/live.py index ffac0af..e8001ef 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -212,6 +212,13 @@ class Interpolator: gen_in['y'] = lerp_label.output self.gen_img = generator(gen_in, signature=gen_signature) + gen_layer_name = 'module_apply_' + gen_signature + '/' + params.inv_layer + gen_encoding = tf.get_default_graph().get_tensor_by_name(gen_layer_name) + ENC_SHAPE = gen_encoding.get_shape().as_list()[1:] + set_encoding = tf.get_variable(name='encoding', dtype=tf.float32, shape=[BATCH_SIZE,] + ENC_SHAPE) + encoding_mix = LerpParam('encoding_mix', a_in=gen_encoding, b_in=set_encoding, shape=[BATCH_SIZE,] + ENC_SHAPE, datatype="input") + tf.contrib.graph_editor.swap_ts(gen_encoding, encoding_mix) + sys.stderr.write("Sin params: {}\n".format(", ".join(self.sin_params.keys()))) sys.stderr.write("Lerp params: {}\n".format(", ".join(self.lerp_params.keys()))) sys.stderr.write("Opts: {}\n".format(", ".join(self.opts.keys()))) |