summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-09 01:01:33 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-09 01:01:33 +0100
commit5ee9167828d2fa405678352b234d5d26b4e453da (patch)
tree089b3204b698b8cd609f7a416d5d54cda9e953b2 /cli/app/search/search_dense.py
parentb9f515173f480a0468fc54d05bf134ebb15b0328 (diff)
folder id :(
Diffstat (limited to 'cli/app/search/search_dense.py')
-rw-r--r--cli/app/search/search_dense.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index 5bf77fc..e006fdd 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -428,6 +428,7 @@ def find_dense_embedding_for_images(params):
fp_out_pkl = os.path.join(app_cfg.DIR_VECTORS, "file_{}.pkl".format(file_id))
out_data = {
'id': file_id,
+ 'folder_id': params.folder_id,
'sample_fn': sample_fn,
'label': out_labels[out_i],
'latent': out_lat[out_i],