summaryrefslogtreecommitdiff
path: root/megapixels/app/processors
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-01-12 19:02:19 +0100
committerJules Laplace <julescarbon@gmail.com>2019-01-12 19:02:19 +0100
commit384be7d882d1402220b10bd5b2d0037226b41785 (patch)
tree4d631de49653dd222191f9250ec51a2ed8a9cc2f /megapixels/app/processors
parent521bc1e5251105f895dd88a38ddc889cbd0e4431 (diff)
returning face results again
Diffstat (limited to 'megapixels/app/processors')
-rw-r--r--megapixels/app/processors/faiss.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/megapixels/app/processors/faiss.py b/megapixels/app/processors/faiss.py
index 5156ad71..ab067fd0 100644
--- a/megapixels/app/processors/faiss.py
+++ b/megapixels/app/processors/faiss.py
@@ -30,6 +30,9 @@ def build_faiss_database(name, recipe):
vec_fn = os.path.join(cfg.DIR_FAISS_METADATA, name, "vecs.csv")
index_fn = os.path.join(cfg.DIR_FAISS_INDEXES, name + ".index")
+ if not os.path.exists(vec_fn):
+ return
+
index = faiss.index_factory(recipe.dim, recipe.factory_type)
keys, rows = load_csv_safe(vec_fn)