diff options
Diffstat (limited to 'inversion/live.py')
| -rw-r--r-- | inversion/live.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/inversion/live.py b/inversion/live.py index 23af747..50ac25f 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -88,7 +88,7 @@ def normal_z_sample(shape=(BATCH_SIZE, Z_DIM,)): class SinParam: def __init__(self, name, shape, datatype="noise"): noise = LerpParam(name + '_noise', shape=shape, datatype=datatype) - orbit_radius = InterpolatorParam(name=name + '_radius', value=0.1) + orbit_radius = InterpolatorParam(name=name + '_radius', value=0.25) orbit_speed = InterpolatorParam(name=name + '_speed', value=1.0) orbit_time = InterpolatorParam(name=name + '_time', value=0.0) output = tf.math.sin(orbit_time.variable + noise.output) * orbit_radius.variable @@ -110,7 +110,7 @@ class LerpParam: a = InterpolatorParam(name=name + '_a', shape=shape, datatype=datatype) b = InterpolatorParam(name=name + '_b', shape=shape, datatype=datatype) n = InterpolatorParam(name=name + '_n', value=0.0) - speed = InterpolatorParam(name=name + '_speed', value=0.1) + speed = InterpolatorParam(name=name + '_speed', value=1.0) output = a.variable * (1 - n.variable) + b.variable * n.variable interpolator.lerp_params[name] = self self.name = name @@ -287,7 +287,11 @@ class Listener: self.rpc_client.send_status('processing', True) dt = 1 / FPS for i in range(99999): - if (i % 100) == 0 or i == 1: + if i == 0: + print("Loading network") + elif i == 1: + print("Processing") + elif (i % 100) == 0: print("Step {}".format(i)) gen_time = time.time() gen_images = interpolator.on_step(i, dt, self.sess) |