From a4dfcacbfd87f0909d63df4f938cd69e2db0eaff Mon Sep 17 00:00:00 2001 From: LinasKo Date: Tue, 3 Dec 2024 13:12:14 +0200 Subject: [PATCH] detections __getitem__: earlier check for empty detections --- 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 e4320a16..14f4c0ef 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -1201,10 +1201,10 @@ class Detections: """ if isinstance(index, str): return self.data.get(index) - if isinstance(index, int): - index = [index] if self.is_empty(): return Detections.empty() + if isinstance(index, int): + index = [index] return Detections( xyxy=self.xyxy[index], mask=self.mask[index] if self.mask is not None else None,