ColorJitter

class torchvision.transforms.ColorJitter(brightness: Union[float, Tuple[float, float]] = 0, contrast: Union[float, Tuple[float, float]] = 0, saturation: Union[float, Tuple[float, float]] = 0, hue: Union[float, Tuple[float, float]] = 0)

随机改变图像的亮度、对比度、饱和度和色调。如果图像是 torch Tensor，则应具有 […, 1 或 3, H, W] 形状，其中 … 表示任意数量的前导维度。如果 img 是 PIL Image，则不支持模式“1”、“I”、“F”以及带有透明度（alpha 通道）的模式。

参数:

• brightness (float 或 tuple of python:float (min, max)) – 抖动亮度的程度。brightness_factor 从 [max(0, 1 - 亮度), 1 + 亮度] 或给定的 [min, max] 中均匀选择。应为非负数。

• contrast (float 或 tuple of python:float (min, max)) – 抖动对比度的程度。contrast_factor 从 [max(0, 1 - 对比度), 1 + 对比度] 或给定的 [min, max] 中均匀选择。应为非负数。

• saturation (float 或 tuple of python:float (min, max)) – 抖动饱和度的程度。saturation_factor 从 [max(0, 1 - 饱和度), 1 + 饱和度] 或给定的 [min, max] 中均匀选择。应为非负数。

• hue (float 或 tuple of python:float (min, max)) – 抖动色调的程度。hue_factor 从 [-色调, 色调] 或给定的 [min, max] 中均匀选择。应具有 0<= 色调 <= 0.5 或 -0.5 <= min <= max <= 0.5。要抖动色调，输入图像的像素值必须为非负数，以便转换为 HSV 空间；因此，如果您将图像归一化为具有负值的区间，或者在使用此函数之前使用生成负值的插值，则此函数不起作用。

使用示例 ColorJitter

变换的图示

变换的图示

forward(img)

参数:

img (PIL Image 或 Tensor) – 输入图像。

返回:

颜色抖动后的图像。

返回类型:

PIL Image 或 Tensor

static get_params(brightness: Optional[List[float]], contrast: Optional[List[float]], saturation: Optional[List[float]], hue: Optional[List[float]]) → Tuple[Tensor, Optional[float], Optional[float], Optional[float], Optional[float]]

获取要应用于图像的随机变换的参数。

参数:

• brightness (tuple of python:float (min, max), optional) – 从中均匀选择 brightness_factor 的范围。传递 None 以关闭变换。

• contrast (tuple of python:float (min, max), optional) – 从中均匀选择 contrast_factor 的范围。

• saturation (tuple of python:float (min, max), optional) – 从中均匀选择 saturation_factor 的范围。

• hue (tuple of python:float (min, max), optional) – 从中均匀选择 hue_factor 的范围。

返回:

用于应用随机变换的参数及其随机顺序。

返回类型:

元组