nvidia.dali.fn.brightness_contrast#

nvidia.dali.fn.brightness_contrast(__input, /, *, brightness=1.0, brightness_shift=0.0, bytes_per_sample_hint=[0], contrast=1.0, contrast_center=0.5, dtype=None, preserve=False, device=None, name=None)#

调整图像的亮度和对比度。

亮度和对比度基于以下公式进行调整

out = brightness_shift * output_range +
      brightness * (contrast_center + contrast * (in - contrast_center))

其中 output_range 对于浮点输出为 1，对于整数类型为最大正值。

此操作符还可以更改数据类型。

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

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

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

关键词参数:

  • brightness (float 或 float 的 TensorList，可选，默认值 = 1.0) –

    亮度乘数。

    支持 per-frame 输入。

  • brightness_shift (float 或 float 的 TensorList，可选，默认值 = 0.0) –

    亮度偏移。

    对于有符号类型，1.0 表示该类型可以表示的最大正值。

    支持 per-frame 输入。

  • bytes_per_sample_hint (int 或 int 列表，可选，默认值 = [0]) –

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

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

  • contrast (float 或 float 的 TensorList，可选，默认值 = 1.0) –

    对比度乘数，其中 0.0 产生均匀的灰色。

    支持 per-frame 输入。

  • contrast_center (float 或 float 的 TensorList，可选，默认值 = 0.5) –

    不受对比度影响的强度级别。

    这是当对比度为零时所有像素假定的值。 如果未设置，则使用输入类型正范围的一半（或 float 的 0.5）。

    支持 per-frame 输入。

  • dtype (nvidia.dali.types.DALIDataType, 可选) –

    输出数据类型。

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

  • preserve (bool，可选，默认值 = False) – 阻止操作符从图中删除，即使其输出未使用。

另请参阅

在此页上