From fbfee0a2bb436a6205d67f561dbd6284621504d6 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Wed, 20 Sep 2023 16:55:50 +0200 Subject: move embedder to module --- src/embedders/imgbeddings-api.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 src/embedders/imgbeddings-api.py (limited to 'src/embedders/imgbeddings-api.py') diff --git a/src/embedders/imgbeddings-api.py b/src/embedders/imgbeddings-api.py new file mode 100644 index 0000000..0c890b5 --- /dev/null +++ b/src/embedders/imgbeddings-api.py @@ -0,0 +1,18 @@ +from PIL import Image +from imgbeddings import imgbeddings +import sys, os +import json as j + +b = imgbeddings() + +paths = sys.argv[1:] +batch_size = 8 + +for i in range(0, len(paths), batch_size): + fs = paths[i:i+batch_size] + + ims = [Image.open(open(f, "rb")) for f in fs] + for emb in b.to_embeddings(ims).tolist(): + print(j.dumps(emb)) + +sys.stderr.write("\n") -- cgit v1.2.3-70-g09d2