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 保留的元素的索引，按得分降序排列
- 返回型別:
張量
