Fix documentation for NMS (#1614)

Summary: Pull Request resolved: #1682

Reviewed By: cpuhrsch

Differential Revision: D19156986

Pulled By: fmassa

fbshipit-source-id: 15c945acf9b4e44d62cc6b0252efe7c143ce570d

Author: fmassa

torchvision/ops/boxes.py

@@ -19,7 +19,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
     -------
@@ -49,7 +49,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
     -------