Загрузка набора данных изображений в numpy массивах в python

Question

У меня 6 разных изображений. Я хочу хранить их вместе в одном массиве numpy. Это возможно? Если да, как я могу это сделать?

from PIL import Image
from matplotlib import image
import matplotlib.pyplot as plt
from os import listdir
from numpy import asarray
import numpy as np

for i in range(1,6):
    image=Image.open(str(i)+'.jpg')
    image=image.resize((100,100))
    temp=asarray(image)
    print(np.append(X_train,temp,axis=0))

Это вызывает следующее исключение:

ValueError: все входные массивы должны иметь одинаковое количество измерений

Answer 1

вы можете создать список массивов и преобразовать обратно в numpy массив

list_of_pics = list()
for i in range(1,6):
    image=Image.open(str(i)+'.jpg')
    image=image.resize((100,100))
    list_of_pics.append(np.asarray(image))
new_array = np.array(list_of_pics)

окончательные размеры new_array должны быть (6 100 100)