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, optional, default = [0]) –

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

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

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