Можете ли вы передать свои собственные веса для Skimage.color.rgb2gray?

Question

Страница документации для skimage.color.rgb2gray гласит:

Веса, использованные в этом преобразовании, откалиброваны для современных люминофорных электронно-лучевых трубок:

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