From 4dde484141ed1a67cd611bd448f0fa2c2757be93 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Wed, 8 Mar 2023 11:31:18 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=96=A4=20make=20black=20happy?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/detection/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/supervision/detection/utils.py b/supervision/detection/utils.py index f42ac26e..5d8407f4 100644 --- a/supervision/detection/utils.py +++ b/supervision/detection/utils.py @@ -46,7 +46,9 @@ def box_iou_batch(boxes_true: np.ndarray, boxes_detection: np.ndarray) -> np.nda return area_inter / (area_true[:, None] + area_detection - area_inter) -def non_max_suppression(predictions: np.ndarray, iou_threshold: float = 0.5) -> np.ndarray: +def non_max_suppression( + predictions: np.ndarray, iou_threshold: float = 0.5 +) -> np.ndarray: """ Perform Non-Maximum Suppression (NMS) on object detection predictions.