From 4b89bece4fdc849a7ed7f0e1e1bf49bb1c12f17e Mon Sep 17 00:00:00 2001 From: Linas Kondrackis Date: Wed, 24 Apr 2024 14:58:52 +0300 Subject: [PATCH] Fix keypoint crash with no detections --- supervision/keypoint/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/keypoint/core.py b/supervision/keypoint/core.py index aca50ad5..6749f845 100644 --- a/supervision/keypoint/core.py +++ b/supervision/keypoint/core.py @@ -135,7 +135,7 @@ class KeyPoints: keypoints = sv.KeyPoints.from_ultralytics(result) ``` """ - if len(ultralytics_results.keypoints.xy) == 0: + if ultralytics_results.keypoints.xy.numel() == 0: return cls.empty() xy = ultralytics_results.keypoints.xy.cpu().numpy()