facebook
/

detr-resnet-101

Object Detection

Model card Files Files and versions

nielsr HF Staff commited on Oct 17, 2023

Commit

4b74f27

·

1 Parent(s): 05d4ab5

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -45,8 +45,9 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101")
-model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101")
 inputs = processor(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)
+# you can specify the revision tag if you don't want the timm dependency
+processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101", revision="no_timm")
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101", revision="no_timm")
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)