diff --git a/groundingdino/util/inference.py b/groundingdino/util/inference.py index 44f7325..84a962e 100644 --- a/groundingdino/util/inference.py +++ b/groundingdino/util/inference.py @@ -121,7 +121,7 @@ def annotate(image_source: np.ndarray, boxes: torch.Tensor, logits: torch.Tensor 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) annotated_frame = cv2.cvtColor(image_source, cv2.COLOR_RGB2BGR) annotated_frame = bbox_annotator.annotate(scene=annotated_frame, detections=detections) diff --git a/requirements.txt b/requirements.txt index a64b8a5..24aa11d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,5 +6,5 @@ yapf timm numpy opencv-python -supervision>=0.15.0 +supervision>=0.22.0 pycocotools