ищет способ сложить изображения в 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)