nvidia.dali.experimental.dynamic.equalize#

nvidia.dali.experimental.dynamic.equalize(input, /, *, batch_size=None, device=None)#

Performs grayscale/per-channel histogram equalization.

The supported inputs are images and videos of uint8_t type.

This operator allows sequence inputs.

Supported backends
  • ‘cpu’

  • ‘gpu’

Parameters:

input (Tensor/Batch ('HW', 'HWC', 'CHW', 'FHW', 'FHWC', 'FCHW')) – Input to the operator.

Keyword Arguments:
  • batch_size (int, optional) – The batch size to broadcast input tensors to. Ignored for batch inputs.

  • device (device-like, optional) – The device to use for the operation. Must not conflict with the device of the inputs.