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