地瓜官网
地瓜开发者社区
地瓜官网
地瓜开发者社区

nn.BgrToYuv444

class horizon_plugin_pytorch.nn.BgrToYuv444 (channel_reversal: bool = False)

Convert image color format from bgr to yuv444.

Parameters: channel_reversal (bool) – Color channel order, set to True when used on RGB input. Defaults to False.

forward (input: Tensor)

Forward pass of BgrToYuv444.

Parameters: input ( Tensor)