megengine.functional.vision.cvt_color

cvt_color(inp, mode='')

将图片从一种颜色空间转换到另一种颜色空间

参数

• inp (Tensor) – 输入图片

• mode (str) – 颜色空间

返回

转换结果

例如：

import numpy as np
import megengine as mge
import megengine.functional as F

x = mge.tensor(np.array([[[[-0.58675045, 1.7526233, 0.10702174]]]]).astype(np.float32))
y = F.vision.cvt_color(x, mode="RGB2GRAY")
print(y.numpy())

输出：

[[[[0.86555195]]]]

megengine.functional.distributed.reduce_scatter_sum megengine.functional.vision.interpolate