microsoft
/

swin-large-patch4-window12-384-in22k

Image Classification

Model card Files Files and versions

nielsr HF Staff commited on Jan 28, 2022

Commit

a089dd5

·

1 Parent(s): f5c5a23

Fix code example

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,8 +45,8 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/swin-tiny-patch4-window7-224")
-model = SwinForImageClassification.from_pretrained("microsoft/swin-tiny-patch4-window7-224")
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)

 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
+feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/swin-large-patch4-window12-384-in22k")
+model = SwinForImageClassification.from_pretrained("microsoft/swin-large-patch4-window12-384-in22k")
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)