image-vectors makes extracting embedding vectors from images as simple as possible.
Install image-vectors from PyPI:
pip install image-vectorsand start extracting embedding vectors:
from PIL import Image
from image_vectors import ImageEmbedder
embedder = ImageEmbedder(model="vit_base_patch32_clip_224.openai")
image = Image.open(<path_to_image>)
embedding = embedder.embed(image)| Model | Embedding Dimensions |
|---|---|
vit_base_patch32_clip_224.openai |
768 |
vit_base_patch16_clip_224.openai |
768 |
| Model | Embedding Dimensions |
|---|---|
vit_small_patch14_dinov2.lvd142m |
384 |
| Model | Embedding Dimensions |
|---|---|
samvit_base_patch16.sa1b |
256 |