numpy的二维数组和三维数组合并

项目中希望把一个RGB图和gray图进行合并，然后进行后续的数据处理，如下方式可实现两图之间的合并

 

import cv2
import numpy as np
from PIL import Image
from torch.utils.data.dataset import Dataset


h,w =10,10

new_image = np.zeros([h, w, 3])
new_image_gray = np.zeros([h, w])


img_merge = np.concatenate((new_image, new_image_gray[:,:,None]), axis=2)
print(img_merge.shape)

cv2.imwrite("filename2.jpg", img_merge)

new_imge的数据如下：

new_image_gray的数据如下：

 

img_merge的数据如下：