快捷方式

ColorJitter

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)[source]

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

如果输入是 torch.Tensor,则预期其形状为 […, 1 或 3, H, W],其中 … 表示任意数量的前导维度。如果 img 是 PIL Image,则不支持模式“1”、“I”、“F”和具有透明度(alpha 通道)的模式。

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

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

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

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

使用 ColorJitter 的示例

变换的图示

变换的图示

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

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

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

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

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

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

返回:

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

返回类型:

元组

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源