nvidia.dali.fn.color_twist#

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

调整图像的色调、饱和度、亮度及对比度。

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

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

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

关键字参数:

  • brightness (float 或 float 的 TensorList, optional, default = 1.0) –

    亮度变化因子。

    值必须为非负数。

    示例值

    • 0 - 黑色图像。

    • 1 - 无变化。

    • 2 - 亮度增加两倍。

    支持 per-frame 输入。

  • bytes_per_sample_hint (int 或 int 的列表, optional, default = [0]) –

    输出大小提示，以每样本字节为单位。

    如果指定，则运算符驻留在 GPU 或分页锁定主机内存中的输出将被预先分配，以适应此大小的样本批次。

  • contrast (float 或 float 的 TensorList, optional, default = 1.0) –

    对比度变化因子。

    值必须为非负数。

    示例值

    • 0 - 均匀灰色图像。

    • 1 - 无变化。

    • 2 - 亮度增加两倍。

    支持 per-frame 输入。

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

    输出数据类型。

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

  • hue (float 或 float 的 TensorList, optional, default = 0.0) –

    色调变化，以度为单位。

    支持 per-frame 输入。

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

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

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

    饱和度变化因子。

    值必须为非负数。

    示例值

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

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

    支持 per-frame 输入。

在本页中