From 472d27841853036a6bd7da01613cddb61c68f0a6 Mon Sep 17 00:00:00 2001 From: Sarath Kumar Mannam <94332524+saruCRCV@users.noreply.github.com> Date: Tue, 19 Apr 2022 18:28:04 -0400 Subject: [PATCH] Update nms.py Add few comments --- ML/Pytorch/object_detection/metrics/nms.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ML/Pytorch/object_detection/metrics/nms.py b/ML/Pytorch/object_detection/metrics/nms.py index 6492f463..d96f02ff 100644 --- a/ML/Pytorch/object_detection/metrics/nms.py +++ b/ML/Pytorch/object_detection/metrics/nms.py @@ -8,10 +8,11 @@ def nms(bboxes, iou_threshold, threshold, box_format="corners"): Parameters: bboxes (list): list of lists containing all bboxes with each bboxes specified as [class_pred, prob_score, x1, y1, x2, y2] - iou_threshold (float): threshold where predicted bboxes is correct - threshold (float): threshold to remove predicted bboxes (independent of IoU) + iou_threshold (nms)(float): threshold where predicted bboxes is correct + threshold (objectness_threshold) (float): threshold to remove predicted bboxes (independent of IoU) box_format (str): "midpoint" or "corners" used to specify bboxes - + iou_threshold is nms threshold;