颜色抖动

class torchvision.transforms.v2.ColorJitter(brightness: Optional[Union[float, Sequence[float]]] = None, contrast: Optional[Union[float, Sequence[float]]] = None, saturation: Optional[Union[float, Sequence[float]]] = None, hue: Optional[Union[float, Sequence[float]]] = None)

随机更改图像或视频的亮度、对比度、饱和度和色调。

如果输入是torch.Tensor，则期望其具有[…, 1 或 3, H, W]形状，其中…表示任意数量的前导维度。如果img是PIL图像，则不支持模式“1”、“I”、“F”以及具有透明度（alpha通道）的模式。

Parameters:

• 亮度 (浮点数 或 python:浮点数的元组 (最小值, 最大值)) – 亮度抖动的程度。 brightness_factor 从 [max(0, 1 - brightness), 1 + brightness] 或给定的 [min, max] 中均匀选择。应为非负数。

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

• 饱和度 (浮点数 或 python:浮点数的元组 (最小值, 最大值)) – 饱和度抖动的程度。 saturation_factor 从 [max(0, 1 - saturation), 1 + saturation] 或给定的 [min, max] 中均匀选择。应为非负数。

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

使用 ColorJitter 的示例：

变换的图示

Illustration of transforms

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]]

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

Parameters:

• brightness (python:float 的元组 (最小值, 最大值), 可选) – 从中均匀选择 brightness_factor 的范围。传递 None 以关闭此变换。

• 对比度 (python:float 的元组 (最小值, 最大值), 可选) – 从中均匀选择对比度因子的范围。传递 None 以关闭此变换。

• saturation (tuple of python:float (min, max), optional) – 饱和度因子的选择范围是均匀的。传递 None 以关闭此变换。

• hue (python:float 的元组 (最小值, 最大值), 可选) – 从中均匀选择 hue_factor 的范围。 传递 None 以关闭变换。

Returns:

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

Return type:

tuple