tfm.vision.iou_similarity.iou

Computes pairwise intersection-over-union between box collections.

gt_boxes a float Tensor with [N, 4].
boxes a float Tensor with [M, 4].

a Tensor with shape [N, M] representing pairwise iou scores.