diff options
| author | jules@lens <julescarbon@gmail.com> | 2019-12-18 23:46:47 +0100 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2019-12-18 23:46:47 +0100 |
| commit | 937c7d6431d6a990b959aabb5fb2e65824fcd4c0 (patch) | |
| tree | 61b12ef086a598eaf24b66d935f3de34b1467379 | |
| parent | ba4d18a2d20eed034ea7926d10e5f760d3809ba2 (diff) | |
fix
| -rw-r--r-- | inversion/live.py | 9 | ||||
| -rw-r--r-- | 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): |