generalized_box_iou¶
- torchvision.ops.generalized_box_iou(boxes1: Tensor, boxes2: Tensor) Tensor [source]¶
Return generalized intersection-over-union (Jaccard index) between two sets of boxes.
Both sets of boxes are expected to be in
(x1, y1, x2, y2)
format with0 <= x1 < x2
and0 <= y1 < y2
.- Parameters:
boxes1 (Tensor[N, 4]) – first set of boxes
boxes2 (Tensor[M, 4]) – second set of boxes
- Returns:
the NxM matrix containing the pairwise generalized IoU values for every element in boxes1 and boxes2
- Return type:
Tensor[N, M]