rgb2bgr

horizon_plugin_pytorch.rgb2bgr(input: Tensor)

Convert color space.

Convert images from RGB format to BGR

  • Parameters: input – image in RGB format with shape [N, 3, H, W]
  • Returns: image in BGR format with shape [N, 3, H, W]
  • Return type: Tensor