View source on GitHub |
Compute pairwise intersection areas between boxes.
tfm.vision.iou_similarity.intersection(
gt_boxes, boxes
)
B: batch_size N: number of groundtruth boxes. M: number of anchor boxes.
Args | |
---|---|
gt_boxes
|
a float Tensor with [N, 4], or [B, N, 4] |
boxes
|
a float Tensor with [M, 4], or [B, M, 4] |
Returns | |
---|---|
a float Tensor with shape [N, M] or [B, N, M] representing pairwise intersections. |