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 的重叠边界框

返回:

由 NMS 选出的元素的 int64 张量，按得分降序排序

返回类型:

张量