Merge pull request #146 from stellasphere/patch-2

Update to support newer versions of Supervision
autodistill · May 17, 2024 · bc82593 · bc82593
2 parents d73eb82 + 2c3b8ea
commit bc82593
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autodistill/utils.py b/autodistill/utils.py
@@ -62,13 +62,13 @@ def plot(image: np.ndarray, detections, classes: List[str], raw=False):
     if detections.mask is not None:
         annotator = sv.MaskAnnotator()
     else:
-        annotator = sv.BoxAnnotator()
+        annotator = sv.BoundingBoxAnnotator()
 
     label_annotator = sv.LabelAnnotator()
 
     labels = [
         f"{classes[class_id]} {confidence:0.2f}"
-        for _, _, confidence, class_id, _ in detections
+        for _, _, confidence, class_id, _, _ in detections
     ]
 
     annotated_frame = annotator.annotate(scene=image.copy(), detections=detections)