diff options
Diffstat (limited to 'src/embedders/imgbeddings-api.py')
| -rw-r--r-- | src/embedders/imgbeddings-api.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/src/embedders/imgbeddings-api.py b/src/embedders/imgbeddings-api.py deleted file mode 100644 index 0c890b5..0000000 --- a/src/embedders/imgbeddings-api.py +++ /dev/null @@ -1,18 +0,0 @@ -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") |