rgb2yuv

horizon_plugin_pytorch.rgb2yuv (input: Tensor, swing: str = 'studio')

Convert color space.

Convert images from RGB format to YUV444 BT.601

Parameters:

input (Tensor) – input image in RGB format, ranging 0 to 255

swing (str) – “studio” for YUV studio swing (Y: 16 to 235, U, V: 16 to 240). “full” for YUV full swing (Y, U, V: 0 to 255). default is “studio”

Returns: YUV image

Return type: Tensor