nn.BgrToYuv444
class horizon_plugin_pytorch.nn.BgrToYuv444(channel_reversal: bool = False)
Convert image color format from bgr to yuv444.
- Parameters:
channel_reversal – Color channel order,
set to True when used on RGB input. Defaults to False.
forward(input: Tensor)
Forward pass of BgrToYuv444.