modelapi/vit-tiny-fp16-ov-catalog

ViT-Tiny image classification model, a Geti™ build of ViT-Tiny converted to OpenVINO™ IR with FP16 weights. It maps an input image to class scores. Model weights are hosted in the source repository OpenVINO/vit_tiny_cls-fp16-ov.

How to Use

1. Install required packages:

pip install openvino-model-api[huggingface]

2. Run model inference:

import cv2
from model_api.models import Model
from model_api.visualizer import Visualizer

model = Model.from_pretrained("OpenVINO/vit_tiny_cls-fp16-ov")
image = cv2.imread("image.jpg")
result = model(image)
vis = Visualizer().render(image, result)
cv2.imwrite("output.jpg", vis)

Legal information

The original model is distributed under the Apache-2.0 license. More details can be found in the original model card.