data.py 文件源码

def brightness_transform(img, brightness_min=0.93, brightness_max=1.4):
    """Transform input image brightness

    Transform the input image brightness by a factor returned by a unifrom
    distribution with `brightness_min` and `brightness_max` as params

    Args:
        img: `ndarray`, input image
        brightness_min: float, minimum contrast for transformation
        brightness_max: float, maximum contrast for transformation

    Returns:
        `ndarray`, brightness transformed image
    """
    if isinstance(img, (np.ndarray)):
        img = Image.fromarray(img)
    brightness_param = np.random.uniform(brightness_min, brightness_max)
    t_img = ImageEnhance.Brightness(img).enhance(brightness_param)

    return np.array(t_img)