nms

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

根据边界框的交并比 (IoU) 执行非极大值抑制 (NMS)。

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

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

参数：

• boxes (Tensor[N, 4])) – 要对其执行 NMS 的边界框。它们应为 (x1, y1, x2, y2) 格式，且满足 0 <= x1 < x2 和 0 <= y1 < y2。

• scores (Tensor[N]) – 每个边界框的得分

• iou_threshold (float) – 丢弃所有 IoU > iou_threshold 的重叠边界框

返回：

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

返回类型：

Tensor