搜索
horizon_plugin_pytorch.bgr2rgb (input: Tensor)
Convert color space.
Convert images from BGR format to RGB
Parameters: input (Tensor) – image in BGR format with shape [N, 3, H, W]
Returns: image in RGB format with shape [N, 3, H, W]
Return type: Tensor