mindspore.dataset.vision.RandomColorAdjust

class mindspore.dataset.vision.RandomColorAdjust(brightness=(1, 1), contrast=(1, 1), saturation=(1, 1), hue=(0, 0))[源代码]

随机调整输入图像的亮度、对比度、饱和度和色调。

Note

此操作支持通过 Offload 在 Ascend 或 GPU 平台上运行。

参数:

  • brightness (Union[float, Sequence[float]], 可选) - 亮度调整因子。不能为负,默认值:(1, 1)。 如果是浮点数,则从 [max(0, 1-brightness), 1+brightness] 范围内统一选择因子。 如果它是一个序列,则代表是范围 [min, max],从此范围中选择调整因子。

  • contrast (Union[float, Sequence[float]], 可选) - 对比度调整因子。不能为负,默认值:(1, 1)。 如果是浮点数,则从 [max(0, 1-contrast), 1+contrast] 范围内统一选择因子。 如果它是一个序列,则代表是范围 [min, max],从此范围中选择调整因子。

  • saturation (Union[float, Sequence[float]], 可选) - 饱和度调整因子。不能为负,默认值:(1, 1)。 如果是浮点数,则从 [max(0, 1-saturation), 1+saturation] 范围内统一选择因子。 如果它是一个序列,则代表是范围 [min, max],从此范围中选择调整因子。

  • hue (Union[float, Sequence[float]], 可选) - 色调调整因子,默认值:(0, 0)。 如果是浮点数,则代表是范围 [-hue, hue],从此范围中选择调整因子。注意 hue 取值应为[0, 0.5]。 如果它是一个序列,则代表是范围 [min, max],从此范围中选择调整因子。注意取值范围min和max是 [-0.5, 0.5] 范围内的浮点数,并且min小于等于max。

异常:

  • TypeError - 如果 brightness 不是float或Sequence[float]类型。

  • TypeError - 如果 contrast 不是float或Sequence[float]类型。

  • TypeError - 如果 saturation 不是float或Sequence[float]类型。

  • TypeError - 如果 hue 不是float或Sequence[float]类型。

  • ValueError - 如果 brightness 为负数。

  • ValueError - 如果 contrast 为负数。

  • ValueError - 如果 saturation 为负数。

  • ValueError - 如果 hue 不在 [-0.5, 0.5] 范围内。

  • RuntimeError - 如果输入图像的shape不是 <H, W, C>。

支持平台:

CPU

样例:

>>> decode_op = vision.Decode()
>>> transform_op = vision.RandomColorAdjust(brightness=(0.5, 1),
...                                         contrast=(0.4, 1),
...                                         saturation=(0.3, 1))
>>> transforms_list = [decode_op, transform_op]
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns=["image"])