Я строю модель, используя TesnorFlow и Kras.
Форма входных данных (345, 720, 120, 3). Это массив изображений размером 720х120.
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)
Ожидаемая форма ввода совпадает с формой, которую я задаю функции подгонки. В чем здесь проблема?