图片添加颜色,二值化数据变成rgb数据

不使用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.]]]