bgr2gray

horizon_plugin_pytorch.bgr2gray(input: Tensor)

Convert color space.

Convert images from BGR format to gray

  • Parameters: input – input image in BGR format of shape [N, 3, H, W], ranging 0 to 255
  • Returns: gray image of shape [N, 1, H, W], ranging 0 to 255
  • Return type: Tensor