diff options
Diffstat (limited to 'cli/app/search/search_dense.py')
| -rw-r--r-- | cli/app/search/search_dense.py | 1 |
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], |