| from groundingdino.util.inference import load_model, load_image, predict, annotate, Model |
| import cv2 |
|
|
|
|
| CONFIG_PATH = "GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py" |
| CHECKPOINT_PATH = "./groundingdino_swint_ogc.pth" |
| DEVICE = "cuda" |
| IMAGE_PATH = "assets/demo7.jpg" |
| TEXT_PROMPT = "Horse. Clouds. Grasses. Sky. Hill." |
| BOX_TRESHOLD = 0.35 |
| TEXT_TRESHOLD = 0.25 |
|
|
| image_source, image = load_image(IMAGE_PATH) |
| model = load_model(CONFIG_PATH, CHECKPOINT_PATH) |
|
|
| boxes, logits, phrases = predict( |
| model=model, |
| image=image, |
| caption=TEXT_PROMPT, |
| box_threshold=BOX_TRESHOLD, |
| text_threshold=TEXT_TRESHOLD, |
| device=DEVICE, |
| ) |
|
|
| annotated_frame = annotate(image_source=image_source, boxes=boxes, logits=logits, phrases=phrases) |
| cv2.imwrite("annotated_image.jpg", annotated_frame) |
