diff --git a/supervision/detection/core.py b/supervision/detection/core.py index 4df3b560..2e6572d5 100644 --- a/supervision/detection/core.py +++ b/supervision/detection/core.py @@ -1002,7 +1002,7 @@ class Detections: self, threshold: float = 0.5, class_agnostic: bool = False ) -> Detections: """ - Perform non-maximum suppression on the current set + Perform non-maximum suppression on the current set of object detections or segmentation prediction. Args: diff --git a/supervision/detection/utils.py b/supervision/detection/utils.py index e5633423..d581aed3 100644 --- a/supervision/detection/utils.py +++ b/supervision/detection/utils.py @@ -113,7 +113,7 @@ def mask_non_max_suppression( non-maximum suppression. Raises: - AssertionError: If `iou_threshold` is not within the closed + AssertionError: If `iou_threshold` is not within the closed range from `0` to `1`. """ assert 0 <= iou_threshold <= 1, (