facebook
/

detr-resnet-50

Object Detection

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Oct 17, 2023

Commit

557a3b6

•

1 Parent(s): 480370a

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-50")
-model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
 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-50", revision="no_timm")
+model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", revision="no_timm")
 inputs = processor(images=image, return_tensors="pt")
 outputs = model(**inputs)