data.py 文件源码

def contrast_transform(img, contrast_min=0.8, contrast_max=1.2):
    """Transform input image contrast

    Transform the input image contrast by a factor returned by a unifrom
    distribution with `contarst_min` and `contarst_max` as params

    Args:
        img: `ndarray`, input image
        contrast_min: float, minimum contrast for transformation
        contrast_max: float, maximum contrast for transformation

    Returns:
        `ndarray`, contrast enhanced image
    """
    if isinstance(img, (np.ndarray)):
        img = Image.fromarray(img)
    contrast_param = np.random.uniform(contrast_min, contrast_max)
    t_img = ImageEnhance.Contrast(img).enhance(contrast_param)

    return np.array(t_img)