正在寻找在PythonNumpy中为dataset geneneration堆叠图像的方法

Question

我正在尝试堆叠numpy数组（图像）。我有大约1000个图像每个（20,20），我想把所有图像堆叠成一个数组。目前我正在使用np.dstack。它给了我一个数组，比如（20000，201000）。我希望结果是（1000,20,20）

Answer 1

通过使用^{}，可以指定阵列堆叠的轴

文档中的示例：

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