View source on GitHub |
Computes pairwise intersection-over-union between box collections.
tfm.vision.iou_similarity.iou(
gt_boxes, boxes
)
Args | |
---|---|
gt_boxes
|
a float Tensor with [N, 4]. |
boxes
|
a float Tensor with [M, 4]. |
Returns | |
---|---|
a Tensor with shape [N, M] representing pairwise iou scores. |