Function pre_process_data

monai/transforms/utils_create_transform_ims.py:271–277

If transform requires 2D data, then convert to 2D by selecting the middle of the last dimension.

(data, ndim, is_map, is_post)

def pre_process_data(data, ndim, is_map, is_post):
    """If transform requires 2D data, then convert to 2D by selecting the middle of the last dimension."""
    if ndim == 2:
        data = {k: v[..., v.shape[-1] // 2] for k, v in data.items()}
    if is_map:
        return data
    return data[CommonKeys.LABEL] if is_post else data[CommonKeys.IMAGE]


def get_2d_slice(image, view: int, is_label):