您可以将自己的权重传递给skimage.color.rgb2gray吗？

Question

skimage.color.rgb2gray的{​​{3}}说：

  

此转换中使用的重量已针对当代CRT荧光粉进行了校准：

     

Y = 0.2125 R + 0.7154 G + 0.0721 B

假设我要使用自己的权重，例如Y = Wr * R + Wg * G + Wb * B。有没有一种方法可以将类似[Wr, Wg, Wb]的数组传递给rgb2gray以便使用它呢？

Answer 1

目前尚无此方法，但是由于通道是数组的最后一个轴，因此使用矩阵乘法实际上很容易做到这一点：

Y = image @ [Wr, Wg, Wb]

因此，您可以轻松地编写自己的代码（或直接使用这种单行代码）。

（注意：在Python 3.4或更早版本中，您应该使用np.dot(image, [Wr, Wg, Wb])。）