Merge pull request #1698 from souhhmm/fix/xyxy-shape-crash

fix: is_empty check for filtering
roboflow · Dec 4, 2024 · ae8e6b2 · ae8e6b2
2 parents 3d14167 + 7141f11
commit ae8e6b2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/supervision/detection/core.py b/supervision/detection/core.py
@@ -1201,6 +1201,8 @@ def __getitem__(
         """
         if isinstance(index, str):
             return self.data.get(index)
+        if self.is_empty():
+            return Detections.empty()
         if isinstance(index, int):
             index = [index]
         return Detections(

diff --git a/test/detection/test_core.py b/test/detection/test_core.py
@@ -223,6 +223,12 @@
             None,
             pytest.raises(IndexError),
         ),
+        (
+            Detections.empty(),
+            np.isin(Detections.empty()["class_name"], ["cat", "dog"]),
+            Detections.empty(),
+            DoesNotRaise(),
+        ),  # Filter an empty detections by specific class names
     ],
 )
 def test_getitem(