box_iou¶
- torchvision.ops.box_iou(boxes1: Tensor, boxes2: Tensor) Tensor [source]¶
返回两组边界框之间的交并比(Jaccard 指数)。
两组边界框都应采用
(x1, y1, x2, y2)
格式,并满足0 <= x1 < x2
和0 <= y1 < y2
的条件。- 参数:
boxes1 (Tensor[N, 4]) – 第一组边界框
boxes2 (Tensor[M, 4]) – 第二组边界框
- 返回值:
一个 NxM 矩阵,包含 boxes1 中每个元素与 boxes2 中每个元素之间的成对 IoU 值
- 返回类型:
Tensor[N, M]