Function convert_points_to_box

monai/transforms/spatial/functional.py:708–723

Converts points to an axis-aligned bounding box. Args: points: Points representing the corners of the bounding box. Shape (N, 8, 3) for the 8 corners of a 3D cuboid or (N, 4, 2) for the 4 corners of a 2D rectangle.

(points)

def convert_points_to_box(points):
    """
    Converts points to an axis-aligned bounding box.

    Args:
        points: Points representing the corners of the bounding box. Shape (N, 8, 3) for the 8 corners of
            a 3D cuboid or (N, 4, 2) for the 4 corners of a 2D rectangle.
    """
    from monai.transforms.utils_pytorch_numpy_unification import max, min

    mins = min(points, dim=1)
    maxs = max(points, dim=1)
    # Concatenate the min and max values to get the bounding boxes
    bboxes = concatenate([mins, maxs], axis=1)

    return bboxes