From 889461574f047891cd274ef1325ed41bfbf47ba0 Mon Sep 17 00:00:00 2001 From: alexandercarruthers <48217309+alexandercarruthers@users.noreply.github.com> Date: Mon, 17 Apr 2023 21:26:04 +0100 Subject: [PATCH] Make Detection docs consistent with __iter__ --- supervision/detection/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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):