diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 14:38:12 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 14:38:12 +0100 |
| commit | 39af2281230604257115757a079fc5d705faecbf (patch) | |
| tree | 4e80ec19257912605f47597a2182bb5f2ae1b38e /cli/app | |
| parent | 405774fa2cc6731d4fea3b48177b4df9693edd4e (diff) | |
fix disentangled
Diffstat (limited to 'cli/app')
| -rw-r--r-- | cli/app/search/live.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 398e25e..9fcf91b 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -272,11 +272,11 @@ class Interpolator: 'rotate2d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk'), 'rotate3d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate3d/model.ckpt'), 'walk'), } - disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'color/model.ckpt'), 'walk')[0] - self.disentangled['r'] = disentangled_color[:, 0] - self.disentangled['g'] = disentangled_color[:, 1] - self.disentangled['b'] = disentangled_color[:, 2] - self.disentangled['luminance'] = np.sum(disentangled_color, axis=1) + disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'color/model.ckpt'), 'walk') + self.disentangled['r'] = disentangled_color[:, :, 0] + self.disentangled['g'] = disentangled_color[:, :, 1] + self.disentangled['b'] = disentangled_color[:, :, 2] + self.disentangled['luminance'] = np.sum(disentangled_color, axis=2) def get_feed_dict(self): |