Unicom-ViT-B-32 / README.md
generall93's picture
unicom-vit-b-32 model card (#1)
0e77bf0 verified
metadata
license: apache-2.0
pipeline_tag: image-feature-extraction

ONNX port of Unicom model from open-metric-learning.

This model is intended to be used for similarity search.

Usage

Here's an example of performing inference using the model with FastEmbed.

from fastembed import ImageEmbedding

images = [
    "./path/to/image1.jpg",
    "./path/to/image2.jpg",
]

model = ImageEmbedding(model_name="Qdrant/Unicom-ViT-B-32")
embeddings = list(model.embed(images))

# [
#   array([ 0.04177791,  0.0550059 ,  0.00025418,  0.0252876 , ..., dtype=float32),
#   array([2.23932182e-03,  4.68995124e-02,  3.28772422e-03,  7.57176951e-02, ...], dtype=float32)
# ]