megengine.data.transform.ColorJitter

class ColorJitter(brightness=0, contrast=0, saturation=0, hue=0, *, order=None)[源代码]

随机改变图像的亮度、对比度、饱和度和色调。

参数
  • brightness – 亮度抖动的程度。从[max(0, 1 - brightness), 1 + brightness]或给定的[min, max]区间上的均匀分布中采样。应为非负数。

  • contrast – 对比度抖动程度。从 [max(0, 1 - contrast), 1 + contrast] 或给定 [min, max] 区间上的均匀分布采样。应为非负数。

  • saturation – 饱和度抖动的程度。从 [max(0, 1 - saturation), 1 + saturation] 或给定 [min, max] 区间上的均匀分布中采样。应为非负数。

  • hue – 色调抖动的程度。从 [-hue, hue] 或给定 [min, max] 区间上的均匀分布中采样。应满足 0<= hue <= 0.5 或 -0.5 <= min <= max <= 0.5.

  • order – 与 VisionTransform 一致。

方法

apply(input)

对单个输入数据实施数据增强。

apply_batch(inputs)

对一批(batch)的输入数据实施数据增广。