Ramos-Ramos
/

dino-resnet-50

Image Feature Extraction

Inference Endpoints

Model card Files Files and versions Community

patrickramos commited on Dec 4, 2022

Commit

149f17e

•

1 Parent(s): 74c9d9a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ 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/resnet-50')
 model = ResNetModel.from_pretrained('Ramos-Ramos/dino-resnet-50')
 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('Ramos-Ramos/dino-resnet-50')
 model = ResNetModel.from_pretrained('Ramos-Ramos/dino-resnet-50')
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)