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
