ошибка дополнения данных в функции генератора подгонки

Question

Я пытаюсь обучить модели cnn для определения выражения лица и уменьшения дисбаланса во входных классах, я использую ImageDataGenerator от keras чтобы увеличить мой набор данных.

это мой код:

xs >> input 

ys >> output

data_gen = ImageDataGenerator(horizontal_flip =True, featurewise_center=True, featurewise_std_normalization=True)

data_gen.fit(xs)

history = self.model.fit_generator((data_gen.flow(xs,ys batch_size=self.batch_size)), steps_per_epoch=len(xs) / self.batch_size, epochs=epochs_num)

Я сгенерировал много изображений и подал их в модель для обучения, но это дает мне эту ошибку: не удалось передать входной массив из формы (28709,128) в форму (28709)

Как я могу определить причину этой ошибки?

Answer 1

Было бы неплохо, если бы вы поделились рабочим примером своего кода, чтобы можно было воспроизвести ошибку.

Мое первое предположение состоит в том, что какая-то переменная (вероятно, вашаbatch_size), то есть 128, портит форму вашего ввода.

Возможно, это можно исправить с помощью reshape из numpy.