Skip to main content

Create image embeddings easily.

Project description

image-vectors

image-vectors makes extracting embedding vectors from images as simple as possible.

Get Started

Install image-vectors from PyPI:

pip install image-vectors

and 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)

Available Embedding Models

CLIP Models

Model Embedding Dimensions
vit_base_patch32_clip_224.openai 768
vit_base_patch16_clip_224.openai 768

DINO Models

Model Embedding Dimensions
vit_small_patch14_dinov2.lvd142m 384

SAM Models

Model Embedding Dimensions
samvit_base_patch16.sa1b 256

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

image_vectors-0.1.1.tar.gz (127.9 kB view hashes)

Uploaded Source

Built Distribution

image_vectors-0.1.1-py3-none-any.whl (2.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page