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 保留的元素的索引，按得分降序排列

返回类型:

张量