diff options
| -rw-r--r-- | megapixels/app/processors/faiss.py | 2 | ||||
| -rw-r--r-- | megapixels/app/server/api.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/megapixels/app/processors/faiss.py b/megapixels/app/processors/faiss.py index ab067fd0..0de8ec69 100644 --- a/megapixels/app/processors/faiss.py +++ b/megapixels/app/processors/faiss.py @@ -27,7 +27,7 @@ def build_all_faiss_databases(): build_faiss_database(name, DefaultRecipe()) def build_faiss_database(name, recipe): - vec_fn = os.path.join(cfg.DIR_FAISS_METADATA, name, "vecs.csv") + vec_fn = os.path.join(cfg.DIR_FAISS_METADATA, name, "face_vector.csv") index_fn = os.path.join(cfg.DIR_FAISS_INDEXES, name + ".index") if not os.path.exists(vec_fn): diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py index 5f80a0c4..48279040 100644 --- a/megapixels/app/server/api.py +++ b/megapixels/app/server/api.py @@ -111,7 +111,7 @@ def upload(dataset_name): for _d, _i in zip(distances, indexes): if _d <= THRESHOLD: dists.append(round(float(_d), 2)) - ids.append(_i+1) + ids.append(_i) identities = [ dataset.get_identity(int(_i)) for _i in ids ] identities = list(filter(None, identities)) |