summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/extract_dense_vectors.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
-rw-r--r--cli/app/commands/biggan/extract_dense_vectors.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py
index 7dc25bc..85e751c 100644
--- a/cli/app/commands/biggan/extract_dense_vectors.py
+++ b/cli/app/commands/biggan/extract_dense_vectors.py
@@ -46,6 +46,7 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video,
tag = "folder_{}".format(folder['id'])
paths = [file['path'] for file in unprocessed_files]
+ opt_feature_layers = opt_feature_layers.split(',')
find_nearest_vector_for_images(
paths=paths,