Объект 'str' не имеет атрибута 'ndim' keras

Question

У меня есть модель Keras, которую я кормлю массивами как X_train, а y_train - это что-то вроде ['0', '1', ...] со строками. Когда я кормлю модель, я получаю сообщение об ошибке выше, но у меня есть эта проблема только с использованием специального генератора изображений, в то время как если я использую класс ImageDataGenerator keras, все в порядке. Есть ли у вас какие-либо предложения о том, что я делаю неправильно?

Это нормально:

aug = ImageDataGenerator(
        width_shift_range=0.2,
        height_shift_range=0.2,
        shear_range=0.15,
        horizontal_flip=True,
        fill_mode="nearest")
def genOK(aug):
    (x, y) = next(aug.flow(np.array(x),y,batch_size=batch_size))
    yield x, y

model.fit_generator(genOK,...)

Это не так:

my_data_augmentation_generator(X,y):
    while True:
        do some image transformations here
        yield X_transformed, y

my_aug = my_data_augmentation_generator()

def genNOTOK(my_aug):
    (x, y) = next(my_aug)
    yield x, y

model.fit_generator(genNOTOK,...)