не удалось передать входной массив из формы (150,150,3) в форму (150,150,3,3)

Question

Я пытался обучить модель

EPOCHS = 5
history = model.fit(train_data, steps_per_epoch=len(train_generator), epochs=EPOCHS, validation_data=val_data, validation_steps=len(val_generator))

, но получаю эту ошибку:

Ошибка значения: не удалось передать входной массив из формы (150,150,3) в форму (150,150,3,3).

Кто-нибудь может мне помочь?

Answer 1

Трансляция будет возможна, если вы измените ввод формы (150,150,3) на (150,150,3,1). например, в тензорном потоке:

`input = tf.expand_dims(input , -1)`