Skip to content

GitLab

Unverified Commit 2e8bcf8b authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix documentation for NMS (#1614)

parent c05da2a8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/ops/boxes.py
View file @ 2e8bcf8b
...@@ -24,7 +24,7 @@ def nms(boxes, scores, iou_threshold): ...@@ -24,7 +24,7 @@ def nms(boxes, scores, iou_threshold):
scores for each one of the boxes scores for each one of the boxes
iou_threshold : float iou_threshold : float
discards all overlapping discards all overlapping
boxes with IoU < iou_threshold boxes with IoU > iou_threshold
Returns Returns
------- -------
...@@ -55,7 +55,7 @@ def batched_nms(boxes, scores, idxs, iou_threshold): ...@@ -55,7 +55,7 @@ def batched_nms(boxes, scores, idxs, iou_threshold):
indices of the categories for each one of the boxes. indices of the categories for each one of the boxes.
iou_threshold : float iou_threshold : float
discards all overlapping boxes discards all overlapping boxes
with IoU < iou_threshold with IoU > iou_threshold
Returns Returns
------- -------
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment