google
/
vit-base-patch16-224
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README

This commit is contained in:
Niels Rogge 2021-03-30 18:13:03 +00:00 committed by huggingface-web
parent 91e26673e9
commit e457400e6d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -41,7 +41,7 @@ url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
image = Image.open(requests.get(url, stream=True).raw) image = Image.open(requests.get(url, stream=True).raw)
feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224') feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224') model = ViTForImageClassification.from_pretrained('google/vit-base-patch16-224')
inputs = feature_extractor(images=image) inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs) outputs = model(**inputs)
logits = outputs.logits logits = outputs.logits
# model predicts one of the 1000 ImageNet classes # model predicts one of the 1000 ImageNet classes