From b3d12ed652f6b27535d36f42614fbfff0939e7ed Mon Sep 17 00:00:00 2001 From: Erol444 Date: Tue, 26 Nov 2024 17:31:28 +0700 Subject: [PATCH] Update core.py Specified that mask is in bool datatype. Used int8 (gpt generated code) and was debugging this issue for 3 hours until I looked into source code of mask annotator --- supervision/detection/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 32753a30a..74c663a89 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -117,7 +117,7 @@ class simplifies data manipulation and filtering, providing a uniform API for 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, H, W)` containing the segmentation masks. + `(n, H, W)` containing the segmentation masks (`bool` data type). 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