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].

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-02-02 UTC.

Args