From 937c7d6431d6a990b959aabb5fb2e65824fcd4c0 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Wed, 18 Dec 2019 23:46:47 +0100 Subject: fix --- inversion/live.py | 9 ++++----- inversion/visualize.py | 2 +- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/inversion/live.py b/inversion/live.py index 8ad38a2..0fdad7c 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -15,14 +15,13 @@ from rpc import CortexRPC from listener import Listener from params import Params -from params_opt import ParamsOpt params = Params('params_dense.json') -opt = ParamsOpt() # -------------------------- # Make directories. # -------------------------- +tag = "test" OUTPUT_DIR = os.path.join('output', tag) if not os.path.exists(OUTPUT_DIR): os.makedirs(OUTPUT_DIR) @@ -74,7 +73,7 @@ class InterpolatorParam: def __init__(self, name, dtype=tf.float32, shape=(), value=None): self.scalar = shape == () self.shape = shape - self.value = np.zeros(shape) if value == None else + self.value = value or np.zeros(shape) self.variable = tf.Variable(self.value, name=name, dtype=dtype, shape=shape) def assign(value): @@ -136,7 +135,7 @@ class Interpolator: def run(cmd, payload): # do things like create a new B and interpolate to it - break + pass class Listener: def __init__(self): @@ -161,7 +160,7 @@ class Listener: self.rpc_client = rpc_client self.rpc_client.send_status('processing', True) for i in range(99999): - gen_images = self.interpolator.on_step(i): + gen_images = self.interpolator.on_step(i) if gen_images is None: break out_img = vs.data2pil(gen_images[0]) diff --git a/inversion/visualize.py b/inversion/visualize.py index 2461458..8bfb38e 100644 --- a/inversion/visualize.py +++ b/inversion/visualize.py @@ -36,7 +36,7 @@ def data2img(data): rescaled = np.clip(rescaled, 0, 255) return np.rint(rescaled).astype('uint8') -def data2pil(data); +def data2pil(data): return Image.fromarray(data2img(data), mode='RGB') def interleave(a, b): -- cgit v1.2.3-70-g09d2