diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 51a24e5b..a8f90576 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -55,15 +55,15 @@ class Detections: Attributes: xyxy (np.ndarray): An array of shape `(n, 4)` containing the bounding boxes coordinates in format `[x1, y1, x2, y2]` mask: (Optional[np.ndarray]): An array of shape `(n, W, H)` containing the segmentation masks. - class_id (Optional[np.ndarray]): An array of shape `(n,)` containing the class ids of the detections. confidence (Optional[np.ndarray]): An array of shape `(n,)` containing the confidence scores of the detections. + class_id (Optional[np.ndarray]): An array of shape `(n,)` containing the class ids of the detections. tracker_id (Optional[np.ndarray]): An array of shape `(n,)` containing the tracker ids of the detections. """ xyxy: np.ndarray mask: np.Optional[np.ndarray] = None - class_id: Optional[np.ndarray] = None confidence: Optional[np.ndarray] = None + class_id: Optional[np.ndarray] = None tracker_id: Optional[np.ndarray] = None def __post_init__(self):