fix docstring about agnostic

ultralytics/yolov5#2162 (comment)
leesongun · Aug 4, 2022 · c3b3945 · c3b3945
1 parent 5bdfde7
commit c3b3945
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cvu/postprocess/nms/yolov5.py b/cvu/postprocess/nms/yolov5.py
@@ -28,7 +28,7 @@ def non_max_suppression_np(predictions: np.ndarray,
         iou_thres (float, optional): IoU threshold in range 0-1 for NMS filtering.
         Defaults to 0.45.
 
-        agnostic (bool, optional): agnostic to width-height. Defaults to False.
+        agnostic (bool, optional): Perform class-agnostic NMS. Defaults to False.
 
         multi_label (bool, optional): apply Multi-Label NMS. Defaults to False.