Fix documentation for NMS (#1614)

2e8bcf8b · Francisco Massa · GitHub · c05da2a8 · 2e8bcf8b
Unverified Commit 2e8bcf8b authored Nov 26, 2019 by Francisco Massa Committed by GitHub Nov 26, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/ops/boxes.py torchvision/ops/boxes.py +2 -2

No files found.
--- a/torchvision/ops/boxes.py
+++ b/torchvision/ops/boxes.py
@@ -24,7 +24,7 @@ def nms(boxes, scores, iou_threshold):
        scores for each one of the boxes
    iou_threshold : float
        discards all overlapping
-        boxes with IoU < iou_threshold
+        boxes with IoU > iou_threshold

    Returns
    -------
@@ -55,7 +55,7 @@ def batched_nms(boxes, scores, idxs, iou_threshold):
        indices of the categories for each one of the boxes.
    iou_threshold : float
        discards all overlapping boxes
-        with IoU < iou_threshold
+        with IoU > iou_threshold

    Returns
    -------