ColorSpaceConversion

class dragon.vm.dali.ops.ColorSpaceConversion(
  image_type,
  output_type,
  **kwargs
)[source]

Convert the color space of image.

Examples:

convert = dali.ops.ColorSpaceConversion('BGR', 'RGB')
y = convert(inputs['x'])

__new__

static ColorSpaceConversion.__new__(
  cls,
  image_type,
  output_type,
  **kwargs
)[source]

Create a ColorSpaceConversion operator.

Parameters:
  • image_type (str) – The color space of input image.
  • output_type (str) – The color space of output image.
Returns:

nvidia.dali.ops.ColorSpaceConversion – The operator.