From acf72b77805ece454dbe17a3736f0808afd92f76 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 12 Jan 2020 03:55:02 +0100 Subject: new checkpoint reader --- cli/app/search/live.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli') diff --git a/cli/app/search/live.py b/cli/app/search/live.py index c2127ab..99bc052 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -94,7 +94,7 @@ disentangled = { 'rotate2d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[0], 'rotate3d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate3d/model.ckpt'), 'walk')[0], } -disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[0] +disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'color/model.ckpt'), 'walk')[0] disentangled['r'] = disentangled_color[:, 0] disentangled['g'] = disentangled_color[:, 1] disentangled['b'] = disentangled_color[:, 2] -- cgit v1.2.3-70-g09d2