src/embedders/imgbeddings-api.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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")