arrays - 无法将输入数组从形状 (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)。

谁能帮我？

标签： arraystensorflowinputmodel

Answer 1

如果将形状 (150,150,3) 的输入更改为 (150,150,3,1) ，则可以进行广播。例如在张量流中：

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