项目中希望把一个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的数据如下: