我正在使用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函数的形状相同。这有什么问题?
答案 0 :(得分:0)
我找到了。这样一个愚蠢的错误。 我在代码中写了input_shape =(170,120,3)而不是(720,120,3)。
在结果中“期望有形状(170,120,3)但是有阵形(720,120,3)”数字170和720对我来说似乎相同?!
模型已成功构建和培训。