Function convert_box_to_points

monai/transforms/spatial/functional.py:658–705

Converts an axis-aligned bounding box to points. Args: mode: The mode specifying how to interpret the bounding box. bbox: Bounding boxes of the shape (N, C) for N boxes. C is [x1, y1, x2, y2] for 2D or [x1, y1, z1, x2, y2, z2] for 3D for each box. Return shape w

(bbox, mode)

def convert_box_to_points(bbox, mode):
    """
    Converts an axis-aligned bounding box to points.

    Args:
        mode: The mode specifying how to interpret the bounding box.
        bbox: Bounding boxes of the shape (N, C) for N boxes. C is [x1, y1, x2, y2] for 2D or [x1, y1, z1, x2, y2, z2]
            for 3D for each box. Return shape will be (N, 4, 2) for 2D or (N, 8, 3) for 3D.

    Returns:
        sequence of points representing the corners of the bounding box.
    """

    mode = get_boxmode(mode)

    points_list = []
    for _num in range(bbox.shape[0]):
        corners = mode.boxes_to_corners(bbox[_num : _num + 1])
        if len(corners) == 4:
            points_list.append(
                concatenate(
                    [
                        concatenate([corners[0], corners[1]], axis=1),
                        concatenate([corners[2], corners[1]], axis=1),
                        concatenate([corners[2], corners[3]], axis=1),
                        concatenate([corners[0], corners[3]], axis=1),
                    ],
                    axis=0,
                )
            )
        else:
            points_list.append(
                concatenate(
                    [
                        concatenate([corners[0], corners[1], corners[2]], axis=1),
                        concatenate([corners[3], corners[1], corners[2]], axis=1),
                        concatenate([corners[3], corners[4], corners[2]], axis=1),
                        concatenate([corners[0], corners[4], corners[2]], axis=1),
                        concatenate([corners[0], corners[1], corners[5]], axis=1),
                        concatenate([corners[3], corners[1], corners[5]], axis=1),
                        concatenate([corners[3], corners[4], corners[5]], axis=1),
                        concatenate([corners[0], corners[4], corners[5]], axis=1),
                    ],
                    axis=0,
                )
            )

    return stack(points_list, dim=0)


def convert_points_to_box(points):