快捷方式

batched_nms

torchvision.ops.batched_nms(boxes: Tensor, scores: Tensor, idxs: Tensor, iou_threshold: float) Tensor[源代码]

以批处理方式执行非最大抑制。

每个索引值对应一个类别,并且不会在不同类别的元素之间应用 NMS。

参数:
  • boxes (Tensor[N, 4]) – 将执行 NMS 的框。它们应采用 (x1, y1, x2, y2) 格式,其中 0 <= x1 < x20 <= y1 < y2

  • scores (Tensor[N]) – 每个框的分数

  • idxs (Tensor[N]) – 每个框的类别的索引。

  • iou_threshold (float) – 丢弃所有 IoU > iou_threshold 的重叠框

返回值:

int64 张量,包含 NMS 保留的元素的索引,按分数降序排序

返回类型:

张量

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源