Make image batch size to encode configurable via config.yml

2026-03-06 05:39:12 +00:00 · 2021-09-16 10:51:39 -07:00
parent 41c328dae0
commit 3afe054312
4 changed files with 9 additions and 8 deletions
--- a/src/main.py
+++ b/src/main.py
@@ -187,8 +187,9 @@ if __name__ == '__main__':
        image_names, image_embeddings, image_metadata_embeddings, image_encoder = image_search.setup(
            pathlib.Path(image_config['input-directory']),
            pathlib.Path(image_config['embeddings-file']),
-            args.regenerate,
-            args.verbose)
+            batch_size=image_config['batch-size'],
+            regenerate=args.regenerate,
+            verbose=args.verbose)

    # Start Application Server
    uvicorn.run(app)