PekingU
/

rtdetr_v2_r50vd

Object Detection

Inference Endpoints

Model card Files Files and versions Community

qubvel-hf HF staff commited on 20 days ago

Commit

c16f17e

·

verified ·

1 Parent(s): bb8d10b

Update README.md (#1)

- Update README.md (e20d9c3a8374f323196e3f8660fa0ede937b75b3)

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -30,8 +30,8 @@ from transformers import RTDetrV2ForObjectDetection, RTDetrImageProcessor
 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
-image_processor = RTDetrImageProcessor.from_pretrained("jadechoghari/rtdetr_v2_r18vd")
-model = RTDetrV2ForObjectDetection.from_pretrained("jadechoghari/rtdetr_v2_r18vd")
 inputs = image_processor(images=image, return_tensors="pt")

 url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
 image = Image.open(requests.get(url, stream=True).raw)
+image_processor = RTDetrImageProcessor.from_pretrained("PekingU/rtdetr_v2_r50vd")
+model = RTDetrV2ForObjectDetection.from_pretrained("PekingU/rtdetr_v2_r50vd")
 inputs = image_processor(images=image, return_tensors="pt")