bgr2centered_yuv

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

Convert color space.

Convert images from BGR format to centered YUV444 BT.601

Parameters:

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

swing (str) – “studio” for YUV studio swing (Y: -112 to 107, U, V: -112 to 112). “full” for YUV full swing (Y, U, V: -128 to 127). default is “studio”

Returns: centered YUV image

Return type: Tensor