Microsoft Swin Tiny Patch4 Window7 224 Ov

🚀 微軟Swin-Tiny Patch4 Window7 224 OpenVINO模型

本項目是將 microsoft/swin-tiny-patch4-window7-224 模型轉換為 OpenVINO 格式，以實現加速推理。

🚀 快速開始

推理示例

以下是對該模型進行推理的示例代碼：

from optimum.intel.openvino import OVModelForImageClassification
from transformers import AutoFeatureExtractor, pipeline

model_id = "helenai/microsoft-swin-tiny-patch4-window7-224"
model = OVModelForImageClassification.from_pretrained(model_id, compile=False)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)

model.reshape(1,3, feature_extractor.size["height"], feature_extractor.size["width"])
model.compile()
ov_pipe = pipeline("image-classification", model=model, feature_extractor=feature_extractor)

image_url_or_path  = "http://images.cocodataset.org/val2017/000000039769.jpg"
ov_pipe(image_url_or_path)