Views
No views yet
1from optimum.intel.openvino import OVModelForImageClassification
2from transformers import AutoFeatureExtractor, pipeline
3
4model_id = "helenai/microsoft-swin-tiny-patch4-window7-224"
5model = OVModelForImageClassification.from_pretrained(model_id, compile=False)
6feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)
7
8model.reshape(1,3, feature_extractor.size["height"], feature_extractor.size["width"])
9model.compile()
10ov_pipe = pipeline("image-classification", model=model, feature_extractor=feature_extractor)
11
12image_url_or_path = "http://images.cocodataset.org/val2017/000000039769.jpg"
13ov_pipe(image_url_or_path)