quoccuongle/grounding-dino
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #342 from ethanlee928/main

Browse Source 
fix Supervision depreciation of BoxAnnotator
This commit is contained in:
Piotr Skalski
2024-07-24 08:59:41 +02:00
committed by GitHub
parent e27a646ca0 e49e881edd
commit 5a890bd867
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
groundingdino/util/inference.py
View File

@@ -121,9 +121,11 @@ def annotate(image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor
in zip(phrases, logits)
]
box_annotator = sv.BoxAnnotator()
bbox_annotator = sv.BoxAnnotator(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 = box_annotator.annotate(scene=annotated_frame, detections=detections, labels=labels)
annotated_frame = bbox_annotator.annotate(scene=annotated_frame, detections=detections)
annotated_frame = label_annotator.annotate(scene=annotated_frame, detections=detections, labels=labels)
return annotated_frame

2
requirements.txt
View File

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