Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "0ce8bcfdaa4ddb58ba1c46c4e9fed1faec315bd6"
Unverified Commit f8f131a3 authored by Fahri Ali Rahman's avatar Fahri Ali Rahman Committed by GitHub
Browse files

Improve documentation for NMS (#2159) 

* Improve documentation for NMS

* update nms doc for special case
parent f9ef235c
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
torchvision/ops/boxes.py
View file @ f8f131a3
...@@ -14,6 +14,11 @@ def nms(boxes, scores, iou_threshold): ...@@ -14,6 +14,11 @@ def nms(boxes, scores, iou_threshold):
IoU greater than iou_threshold with another (higher scoring) IoU greater than iou_threshold with another (higher scoring)
box. box.
If multiple boxes have the exact same score and satisfy the IoU
criterion with respect to a reference box, the selected box is
not guaranteed to be the same between CPU and GPU. This is similar
to the behavior of argsort in PyTorch when repeated values are present.
Parameters Parameters
---------- ----------
boxes : Tensor[N, 4]) boxes : Tensor[N, 4])
......
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