mirrors/GroundingDINO
1
0
Fork 0
mirror of https://github.com/IDEA-Research/GroundingDINO.git synced 2025-06-03 14:50:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

replaced BoundingBoxAnnotator with BoxAnnotator, updated Supervision version

Browse Source
This commit is contained in:
ethanlee928 2024-07-23 23:19:52 +08:00
parent d75c95daf6
commit 8b6a55f612
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
groundingdino/util/inference.py
View File

@ -121,7 +121,7 @@ def annotate(image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor
in zip(phrases, logits) in zip(phrases, logits)
] ]
bbox_annotator = sv.BoundingBoxAnnotator(color_lookup=sv.ColorLookup.INDEX) bbox_annotator = sv.BoxAnnotator(color_lookup=sv.ColorLookup.INDEX)
label_annotator = sv.LabelAnnotator(color_lookup=sv.ColorLookup.INDEX) label_annotator = sv.LabelAnnotator(color_lookup=sv.ColorLookup.INDEX)
annotated_frame = cv2.cvtColor(image_source, cv2.COLOR_RGB2BGR) annotated_frame = cv2.cvtColor(image_source, cv2.COLOR_RGB2BGR)
annotated_frame = bbox_annotator.annotate(scene=annotated_frame, detections=detections) annotated_frame = bbox_annotator.annotate(scene=annotated_frame, detections=detections)

2
requirements.txt
View File

@ -6,5 +6,5 @@ yapf
timm timm
numpy numpy
opencv-python opencv-python
supervision>=0.15.0 supervision>=0.22.0
pycocotools pycocotools