Merge pull request #20 from roboflow/bug/handle_empty_mask_edge_case_…

…on_masks_to_marks Handle empty mask edge case in postprocessing.
roboflow · Dec 4, 2023 · d1dfcd4 · d1dfcd4
2 parents 85b08ab + 2817b53
commit d1dfcd4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/maestro/postprocessing/mask.py b/maestro/postprocessing/mask.py
@@ -183,6 +183,10 @@ def masks_to_marks(masks: np.ndarray) -> sv.Detections:
         sv.Detections: An object containing the masks and their bounding box
             coordinates.
     """
+    if len(masks) == 0:
+        marks = sv.Detections.empty()
+        marks.mask = np.empty((0, 0, 0), dtype=bool)
+        return marks
     return sv.Detections(
         mask=masks,
         xyxy=sv.mask_to_xyxy(masks=masks)