From 3676f07168d30c762ef1d445e5ced55cc4480d21 Mon Sep 17 00:00:00 2001 From: Onuralp SEZER Date: Wed, 26 Jun 2024 15:51:32 +0300 Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9E=20doc=20and=20validation=20?= =?UTF-8?q?typo=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Onuralp SEZER --- supervision/keypoint/core.py | 8 +++++--- supervision/validators/__init__.py | 3 +-- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/supervision/keypoint/core.py b/supervision/keypoint/core.py index f4fd4d0c..facab5f8 100644 --- a/supervision/keypoint/core.py +++ b/supervision/keypoint/core.py @@ -443,13 +443,15 @@ class KeyPoints: predictor = DefaultPredictor(cfg) result = predictor(image) - keypoints = sv.Keypoints.from_detectron2(result) + keypoints = sv.KeyPoints.from_detectron2(result) ``` """ - + if hasattr(detectron2_results["instances"], "pred_keypoints"): return cls( - xy=detectron2_results["instances"].pred_keypoints.cpu().numpy()[:, :, :2], + xy=detectron2_results["instances"] + .pred_keypoints.cpu() + .numpy()[:, :, :2], confidence=detectron2_results["instances"] .pred_keypoints.cpu() .numpy()[:, :, 2:], diff --git a/supervision/validators/__init__.py b/supervision/validators/__init__.py index 9a9fca8c..44cfcb98 100644 --- a/supervision/validators/__init__.py +++ b/supervision/validators/__init__.py @@ -62,8 +62,7 @@ def validate_keypoint_confidence(confidence: Any, n: int, m: int) -> None: if not is_valid: raise ValueError( f"confidence must be a 1D np.ndarray with shape {expected_shape}, but " - "got" - f"shape {actual_shape}" + f"got shape {actual_shape}" )