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]