nvidia.dali.experimental.dynamic.color_twist

nvidia.dali.experimental.dynamic.color_twist(input, /, *, batch_size=None, device=None, brightness=None, contrast=None, dtype=None, hue=None, image_type=None, saturation=None)

Adjusts hue, saturation, brightness and contrast of the image.

This operator allows sequence inputs and supports volumetric data.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

input (Tensor/Batch ('HWC', 'FHWC', 'DHWC')) – Input to the operator.

Keyword Arguments:

• brightness (float or Tensor/Batch of float, optional, default = 1.0) –

  Brightness change factor.

  Values must be non-negative.

  Example values:

  • 0 - Black image.

  • 1 - No change.

  • 2 - Increase brightness twice.

  Supports per-frame inputs.

• contrast (float or Tensor/Batch of float, optional, default = 1.0) –

  Contrast change factor.

  Values must be non-negative.

  Example values:

  • 0 - Uniform grey image.

  • 1 - No change.

  • 2 - Increase brightness twice.

  Supports per-frame inputs.

• dtype (nvidia.dali.types.DALIDataType, optional) –

  Output data type.

  If not set, the input type is used.

• hue (float or Tensor/Batch of float, optional, default = 0.0) –

  Hue change, in degrees.

  Supports per-frame inputs.

• image_type (nvidia.dali.types.DALIImageType, optional, default = DALIImageType.RGB) – The color space of the input and the output image.

• saturation (float or Tensor/Batch of float, optional, default = 1.0) –

  Saturation change factor.

  Values must be non-negative.

  Example values:

  • 0 - Completely desaturated image.

  • 1 - No change to image’s saturation.

  Supports per-frame inputs.