nms

torchvision.ops.nms(boxes: Tensor, scores: Tensor, iou_threshold: float) → Tensor

根据它们的交并比（IoU）对框执行非极大值抑制（NMS）。

NMS 迭代地移除那些与另一个（得分更高的）框的 IoU 大于 iou_threshold 的较低得分框。

如果多个框具有完全相同的分数并且满足与参考框的IoU标准，则在CPU和GPU之间选择的框不能保证相同。这与PyTorch中argsort在存在重复值时的行为类似。

Parameters:

• boxes (Tensor[N, 4])) – 要在其上执行NMS的框。它们 预计为(x1, y1, x2, y2)格式，且0 <= x1 < x2和 0 <= y1 < y2。

• scores (Tensor[N]) – 每个框的分数

• iou_threshold (float) – 丢弃所有与IoU大于iou_threshold的重叠框

Returns:

int64 张量，包含被 NMS 保留的元素的索引，按分数降序排序

Return type:

张量