megengine.functional.vision.cvt_color

cvt_color(inp, mode='')[源代码]

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

参数
  • inp – 输入图片

  • mode – 颜色空间

返回

转换结果

例如:

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]]]]