IDEA-Research
/

grounding-dino-base

Zero-Shot Object Detection

Model card Files Files and versions

nielsr HF Staff commited on Apr 11, 2024

Commit

3c3e0f8

·

verified ·

1 Parent(s): 22fc5fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection,
-model_id = "IDEA-Research/grounding-dino-tiny"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)

 from PIL import Image
 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection,
+model_id = "IDEA-Research/grounding-dino-base"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)