nvidia.dali.fn.contrast#

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

调整图像的对比度。

对比度根据以下公式进行调整

out = contrast_center + contrast * (in - contrast_center)

此运算符还可以更改数据类型。

此运算符允许序列输入并支持体数据。

支持的后端
  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList ('FHWC', 'DHWC', 'HWC')) – 运算符的输入。

关键字参数:
  • 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) – 即使运算符的输出未使用,也阻止从图中删除该运算符。