Keras,方法:适合,检查输入时出错,关于形状

时间:2018-05-06 16:55:54

标签: python input model keras shape

我正在使用TesnorFlow和Kras构建模型。

输入数据的形状是(345,720,120,3)。这是一个大小为720x120的图像数组。

model = Sequential()
model.add(Lambda(lambda x: (x / 255.0) - 0.5, input_shape=(170,120,3)))   
model.add(Conv2D(24,(5,5),strides=(2,2),activation='relu'))
...

model.fit(X_train, y_train, validation_split=0.2, shuffle=True, epochs=4)

最后一行 我收到了一个错误:

ValueError: Error when checking input: expected lambda_1_input to have shape (170, 120, 3) but got array with shape (720, 120, 3)

预期的输入形状与我赋予fit函数的形状相同。这有什么问题?

1 个答案:

答案 0 :(得分:0)

我找到了。这样一个愚蠢的错误。 我在代码中写了input_shape =(170,120,3)而不是(720,120,3)。

在结果中“期望有形状(170,120,3)但是有阵形(720,120,3)”数字170和720对我来说似乎相同?!

模型已成功构建和培训。