diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-01-12 19:02:19 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-01-12 19:02:19 +0100 |
| commit | 384be7d882d1402220b10bd5b2d0037226b41785 (patch) | |
| tree | 4d631de49653dd222191f9250ec51a2ed8a9cc2f /megapixels/app/processors | |
| parent | 521bc1e5251105f895dd88a38ddc889cbd0e4431 (diff) | |
returning face results again
Diffstat (limited to 'megapixels/app/processors')
| -rw-r--r-- | megapixels/app/processors/faiss.py | 3 |
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) |
