def augment_segmentation_map(self, segmentation_map, hooks=None):
    """
    Augment a single segmentation map.

    Parameters
    ----------
    segmentation_map : (H,W,C) ndarray or (H,W) ndarray
        The segmentation map to augment.
        Expected dtypes are integer-likes, e.g. int32.

    hooks : None or ia.HooksImages, optional(default=None)
        HooksImages object to dynamically interfere with the augmentation
        process.

    Returns
    -------
    out : ndarray
        The corresponding augmented segmentation maps.

    """
    assert segmentation_map.ndim in [2, 3], "Expected segmentation map to have shape (height, width, [channels]), got shape %s." % (segmentation_map.shape,)
    return self.augment_segmentation_maps([segmentation_map], hooks=hooks)[0]