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 < x2 且 0 <= y1 < y2。

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

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

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

返回:

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

返回类型:

张量