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