diff options
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()))) |