summaryrefslogtreecommitdiff
path: root/megapixels/commands/faiss/build_faiss.py
diff options
context:
space:
mode:
Diffstat (limited to 'megapixels/commands/faiss/build_faiss.py')
-rw-r--r--megapixels/commands/faiss/build_faiss.py24
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()