Function _yx_in_bounding_boxes

eval/utils/action_utils.py:111–137

Check if the (y,x) point is contained in each bounding box. Args: yx: The (y, x) coordinate in pixels of the point. bounding_boxes: A 2D int array of shape (num_bboxes, 4), where each row represents a bounding box: (y_top_left, x_top_left, box_height, box

(
    yx, bounding_boxes
)

def _yx_in_bounding_boxes(
    yx, bounding_boxes
):
    """Check if the (y,x) point is contained in each bounding box.

    Args:
        yx: The (y, x) coordinate in pixels of the point.
        bounding_boxes: A 2D int array of shape (num_bboxes, 4), where each row
            represents a bounding box: (y_top_left, x_top_left, box_height,
            box_width). Note: containment is inclusive of the bounding box edges.

    Returns:
        is_inside: A 1D bool array where each element specifies if the point is
            contained within the respective box.
    """
    y, x = yx

    # `bounding_boxes` has shape (n_elements, 4); we extract each array along the
    # last axis into shape (n_elements, 1), then squeeze unneeded dimension.
    top, left, height, width = [
        jnp.squeeze(v, axis=-1) for v in jnp.split(bounding_boxes, 4, axis=-1)
    ]

    # The y-axis is inverted for AndroidEnv, so bottom = top + height.
    bottom, right = top + height, left + width

    return jnp.logical_and(y >= top, y <= bottom) & jnp.logical_and(x >= left, x <= right)


def _resize_annotation_bounding_boxes(