summaryrefslogtreecommitdiff
path: root/cli/app/search/live.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli/app/search/live.py')
-rw-r--r--cli/app/search/live.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index 9fcf91b..f066faf 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -207,9 +207,9 @@ class Interpolator:
z_sum = lerp_z.output + sin_z.output
# Latent - saturation
- abs_zoom = InterpolatorParam(name='abs_zoom', value=1.0)
- z_abs = z_sum / tf.abs(z_sum) * abs_zoom.variable
- z_mix = LerpParam('abs_mix', a_in=z_sum, b_in=z_abs, shape=[BATCH_SIZE, Z_DIM], datatype="input")
+ saturation = InterpolatorParam(name='saturation', value=1.0)
+ z_abs = z_sum / tf.abs(z_sum) * saturation.variable
+ z_mix = LerpParam('saturation_mix', a_in=z_sum, b_in=z_abs, shape=[BATCH_SIZE, Z_DIM], datatype="input")
# Latent - disentangled vectors
zoom = InterpolatorParam(name='zoom', value=0.0).variable * self.disentangled['zoom']
@@ -278,7 +278,6 @@ class Interpolator:
self.disentangled['b'] = disentangled_color[:, :, 2]
self.disentangled['luminance'] = np.sum(disentangled_color, axis=2)
-
def get_feed_dict(self):
opt = {}
for param in self.opts.values():