diff options
Diffstat (limited to 'inversion/live.py')
| -rw-r--r-- | inversion/live.py | 8 |
1 files changed, 8 insertions, 0 deletions
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 # -------------------------- |