fix: 🐞 add check if keypoints size is 0 return empty keypoint result

Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
2024-07-11 04:41:05 +03:00 · 2024-07-11 04:41:05 +03:00 · 13fe6e0942
parent 185ae462a4
commit 13fe6e0942
1 changed files with 2 additions and 0 deletions
--- a/supervision/keypoint/core.py
+++ b/supervision/keypoint/core.py
@ -448,6 +448,8 @@ class KeyPoints:
        """

        if hasattr(detectron2_results["instances"], "pred_keypoints"):
+            if detectron2_results["instances"].pred_keypoints.cpu().numpy().size == 0:
+                return cls.empty()
            return cls(
                xy=detectron2_results["instances"]
                .pred_keypoints.cpu()