Merge pull request IDEA-Research#342 from ethanlee928/main

fix Supervision depreciation of BoxAnnotator
Yeongjae · Jul 24, 2024 · 5a890bd · 5a890bd
2 parents e27a646 + e49e881
commit 5a890bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/groundingdino/util/inference.py b/groundingdino/util/inference.py
@@ -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
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -6,5 +6,5 @@ yapf
 timm
 numpy
 opencv-python
-supervision==0.21.0
+supervision>=0.22.0
 pycocotools