summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/search_class.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-06 14:31:12 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-06 14:31:12 +0100
commite5430f310f6bd5430d55517c900bc83317d91048 (patch)
treee0dca7de48dc568852879a6b84862e16455a4108 /cli/app/commands/biggan/search_class.py
parenta11656ea3cb3af5ad58997b2ad8d5e94677284dc (diff)
path
Diffstat (limited to 'cli/app/commands/biggan/search_class.py')
-rw-r--r--cli/app/commands/biggan/search_class.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py
index 28a01fd..bd260fb 100644
--- a/cli/app/commands/biggan/search_class.py
+++ b/cli/app/commands/biggan/search_class.py
@@ -66,7 +66,7 @@ def cli(ctx, opt_fp_in, opt_dims, opt_steps, opt_limit, opt_video, opt_tag):
if index == opt_limit:
break
out_fns[index] = os.path.basename(path)
- fp_frames = find_nearest_vector(sess, path, opt_dims, out_images, out_labels, opt_steps, index)
+ fp_frames = find_nearest_vector(sess, path, opt_dims, out_images, out_labels, out_latent, opt_steps, index)
if opt_video:
export_video(fp_frames)