不使用cv2的方式, 将二值化的数据变为制定的rgb颜色.
rgb = [255,0,100]
t = np.array([[0., 1.],[1., 0.]])
t0 = t.reshape(t.shape[0],t.shape[1],1)
tn = np.concatenate([t0*rgb[0],t0*rgb[1],t0*rgb[2]],axis=2)
print(tn)
输出:
[[[ 0. 0. 0.]
[255. 0. 100.]]
[[255. 0. 100.]
[ 0. 0. 0.]]]