tfm.vision.box_ops.bbox_overlap

Calculates the overlap between proposal and ground truth boxes.

tfm.vision.box_ops.bbox_overlap(
    boxes, gt_boxes
)

Some boxes or gt_boxes may have been padded. The returned iou tensor for these boxes will be -1.

Args
boxes	a tensor with a shape of [batch_size, N, 4]. N is the number of proposals before groundtruth assignment (e.g., rpn_post_nms_topn). The last dimension is the pixel coordinates in [ymin, xmin, ymax, xmax] form.
gt_boxes	a tensor with a shape of [batch_size, MAX_NUM_INSTANCES, 4]. This tensor might have paddings with a negative value.

Returns
iou	a tensor with as a shape of [batch_size, N, MAX_NUM_INSTANCES].