сглаживание нескольких изображений RGB с NumPy

Question

Я пытаюсь объединить несколько изображений в массив (n, 1). Это работает нормально, если я просто загружаю 1 изображение. Однако, если я увеличу количество изображений, это не сработает.

Загрузка изображения

for dataset, path in data_dir.items():
    for i, cat in enumerate(categories):
        cur_dir = os.path.join(path, cat)
        print (" current directory path is :", cur_dir, "\n")
        filenames = glob(os.path.join(cur_dir, '*.png'))
        for file in filenames:
            img = cv.imread(file)
            print (img.shape)
            if dataset == 'Training':
                x_train.append(img)
                y_train.append(i)

Сглаживание изображения

print ("Number of training examples: m_train = " + str(len(y_train)) + " " + str(len(x_train)))
x_train = np.asarray(x_train)
y_train = np.asarray(y_train)

print (x_train.shape[0])

## Logic to Flatten ##
train_set_x_flatten = x_train.reshape(x_train.shape[0],-1).T
train_set_y_flatten = y_train.reshape(y_train.shape[0],-1).T

print ("train_set_x_flatten shape: " + str(train_set_x_flatten.shape))
print ("train_set_y shape: " + str(train_set_y_flatten.shape))

Вывод с 1 изображением :

(297, 319, 3) -- dimension of image
Number of training examples: m_train = 1 1
train_set_x_flatten shape: (284229, 1)
train_set_y shape: (1, 1)

Вывод с 4 изображениями

(297, 319, 3) -- dimension of image
(297, 319, 3) -- dimension of image
(499, 390, 3)
(499, 390, 3)
Number of training examples: m_train = 4 4
train_set_x_flatten shape: (1, 4)
train_set_y shape: (1, 4)