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.