oneflow.nms¶
-
oneflow.
nms
(boxes, scores, iou_threshold: float)¶ Performs non-maximum suppression (NMS) on the boxes according to their intersection-over-union (IoU).
NMS iteratively removes lower scoring boxes which have an IoU greater than iou_threshold with another (higher scoring) box.
- Parameters
- Returns
int64 tensor with the indices of the elements that have been kept by NMS, sorted in decreasing order of scores
- Return type