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