Modelpublic

Qdrant/resnet50-onnx

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face

3likes19kdownloads

Model Card

ONNX port of microsoft/resnet-50.

This model is intended to be used for image classification and similarity searches.

You can find the ONNX port implementation here

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/resnet50-onnx")
embeddings = list(model.embed(images))

# [
#   array([-0.1115,  0.0097,  0.0052,  0.0195, ...], dtype=float32),
#   array([-0.1019,  0.0635, -0.0332,  0.0522, ...], dtype=float32)
# ]