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