nvidia.dali.fn.experimental.equalize

nvidia.dali.fn.experimental.equalize(__input, /, *, bytes_per_sample_hint=[0], preserve=False, device=None, name=None)

执行灰度/每通道直方图均衡化。

支持的输入是 uint8_t 类型的图像和视频。

此操作符允许序列输入。

支持的后端

• ‘cpu’

• ‘gpu’

参数:

__input (TensorList ('HW', 'HWC', 'CHW', 'FHW', 'FHWC', 'FCHW')) – 操作符的输入。

关键字参数:

• bytes_per_sample_hint (int 或 list of int, 可选, 默认 = [0]) –

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

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

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