寻找一种在 python numpy 中堆叠图像以生成数据集的方法

Question

我正在尝试堆叠 numpy 数组（图像）。我每个大约有 1000 张图像（20,20）我想将所有图像堆叠到单个数组中。目前我正在使用np.dstack。它给了我一个像 (20,20,1000) 这样的数组。我希望结果喜欢是 (1000,20,20)。

Answer 1

通过使用numpy.stack，可以指定数组堆叠的轴。

文档中的示例：

>>> arrays = [np.random.randn(3, 4) for _ in range(10)]
>>> np.stack(arrays, axis=0).shape
(10, 3, 4)