图像处理-如何堆叠一个通道的图像以创建2通道阵列

Question

我有两个图像数据集，一个具有900个样本，另一个具有其他900个样本。如果我仅拍摄每个数据集的一张图像并将其转换为数组，则在以灰度模式读取图像后，它们的形状将为（1，128，118）。

如何将这两个数组堆叠为（2，128，118）？

Answer 1

import numpy as np
import cv2
img1 = cv2.imread(...)
img2 = cv2.imread(...)
np.stack([img1,img2],axis=0)