搜索
horizon_plugin_pytorch.bgr2gray (input: Tensor)
Convert color space.
Convert images from BGR format to gray
Parameters: input (Tensor) – 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