ColorJitter¶
- class paddle.vision.transforms. ColorJitter ( brightness=0, contrast=0, saturation=0, hue=0, keys=None ) [源代码] ¶
随机调整图像的亮度、对比度、饱和度和色调。
参数¶
brightness (float,可选) - 亮度调整范围大小,会从给定参数后的均匀分布[max(0,1 - brightness), 1 + brightness]中随机选择进行实际调整,不能是负数。默认值:0。
contrast (float,可选) - 对比度调整范围大小,,会从给定参数后的均匀分布[max(0,1 - contrast), 1 + contrast]中随机选择进行实际调整,不能是负数。默认值:0。
saturation (float,可选) - 饱和度调整范围大小,,会从给定参数后的均匀分布[max(0,1 - saturation), 1 + saturation]中随机选择进行实际调整,不能是负数。默认值:0。
hue (float,可选) - 色调调整范围大小,会从给定参数后的均匀分布[-hue, hue]中随机选择进行实际调整,参数值需要在 0 到 0.5 之间。默认值:0。
keys (list[str]|tuple[str],可选) - 与
BaseTransform
定义一致。默认值:None。
形状¶
img (PIL.Image|np.ndarray|Paddle.Tensor) - 输入的图像数据,数据格式为'HWC'。
output (PIL.Image|np.ndarray|Paddle.Tensor) - 返回调整亮度、对比度、饱和度和色调后的图像数据。
返回¶
计算
ColorJitter
的可调用对象。
代码示例¶
>>> import numpy as np
>>> from PIL import Image
>>> from paddle.vision.transforms import ColorJitter
>>> transform = ColorJitter(0.4, 0.4, 0.4, 0.4)
>>> fake_img = Image.fromarray((np.random.rand(224, 224, 3) * 255.).astype(np.uint8))
>>> fake_img = transform(fake_img)
>>> print(fake_img.size)
(224, 224)