diff options
Diffstat (limited to 'megapixels/commands/faiss/build_faiss.py')
| -rw-r--r-- | megapixels/commands/faiss/build_faiss.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/megapixels/commands/faiss/build_faiss.py b/megapixels/commands/faiss/build_faiss.py new file mode 100644 index 00000000..fc6b37ce --- /dev/null +++ b/megapixels/commands/faiss/build_faiss.py @@ -0,0 +1,24 @@ +""" +Index all of the FAISS datasets +""" + +import os +import glob +import click +import faiss +import time +import numpy as np + +from app.utils.file_utils import load_recipe, load_csv_safe +from app.settings import app_cfg as cfg +from app.processors.faiss import build_all_faiss_databases + +@click.command() +@click.pass_context +def cli(ctx): + """build the FAISS index. + - looks for all datasets in faiss/metadata/ + - uses the recipe above by default + - however you can override this by adding a new recipe in faiss/recipes/{name}.json + """ + build_all_faiss_databases() |
