From b9f515173f480a0468fc54d05bf134ebb15b0328 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 8 Jan 2020 20:17:30 +0100 Subject: moving in the live stuff --- inversion/live.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'inversion') diff --git a/inversion/live.py b/inversion/live.py index e8001ef..6642d0a 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -13,6 +13,7 @@ import time import random from scipy.stats import truncnorm from PIL import Image +from url.parse import parse_qs import visualize as vs tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), '../../live-cortex/rpc/')) @@ -260,6 +261,11 @@ class Interpolator: label[0] /= label[0].sum() self.lerp_params['label'].switch(target_value=label) + def set_encoding(self, opt): + next_id = opt['id'] + should_lerp = opt['lerp'] + + def on_step(self, i, dt, sess): for param in self.sin_params.values(): param.update(dt) @@ -274,6 +280,8 @@ class Interpolator: self.lerp_params[payload].switch() if cmd == 'setCategory': self.set_category(payload) + if cmd == 'setEncoding': + self.set_encoding(parse_qs(payload)) pass # -------------------------- -- cgit v1.2.3-70-g09d2