nvidia.dali.fn.saturation

nvidia.dali.fn.saturation(__input, /, *, bytes_per_sample_hint=[0], dtype=None, image_type=DALIImageType.RGB, preserve=False, saturation=1.0, device=None, name=None)

更改图像的饱和度级别。

此操作符允许序列输入并支持体积数据。

支持的后端

• ‘cpu’

• ‘gpu’

参数:

__input (TensorList ('HWC', 'FHWC', 'DHWC')) – 操作符的输入。

关键字参数:

• bytes_per_sample_hint (int 或 list of int, optional, default = [0]) –

  每个样本的输出大小提示（以字节为单位）。

  如果指定，则将预先分配位于 GPU 或分页锁定主机内存中的操作符输出，以适应此大小的样本批次。

• dtype (nvidia.dali.types.DALIDataType, optional) –

  输出数据类型。

  如果未设置，则使用输入类型。

• image_type (nvidia.dali.types.DALIImageType, optional, default = DALIImageType.RGB) – 输入和输出图像的颜色空间。

• preserve (bool, optional, default = False) – 阻止操作符从图中删除，即使其输出未使用。

• saturation (float 或 TensorList of float, optional, default = 1.0) –

  饱和度变化因子。

  值必须为非负数。

  示例值

  • 0 - 完全去饱和的图像。

  • 1 - 图像饱和度无变化。

  支持 per-frame 输入。