tfm.vision.iou_similarity.area

Stay organized with collections Save and categorize content based on your preferences.

Computes area of boxes.

B: batch_size N: number of boxes

box a float Tensor with [N, 4], or [B, N, 4].

a float Tensor with [N], or [B, N]