def METHOD_NAME(image):
    """
    Returns indices of the maximum values along the channel axis
    Input shape is (bs=1, channel=3, (ROI_SHAPE)), float -- sub-volume inference result
    Output shape is (bs=1, channel=1, (ROI_SHAPE)), integer -- segmentation result
    """
    channel_axis = 1
    image = np.argmax(image, axis=channel_axis).astype(np.uint8)
    image = np.expand_dims(image, axis=0)
    return image