Keras: Получение ошибки «Найдено: Тензор (« input_1: 0 », shape = (None, 256, 256, 2), dtype = float32)» при использовании слоя `Input`

Question

Моя модель:

model = Sequential()
model.add(Input(shape=input_shape))       
model.add(Flatten())

# 2nd Fully Connected Layer
model.add(Dense(120))
model.add(Activation('relu'))

model.add(Dropout(0.3))

# Output Layer
model.add(Dense(n_classes))
model.add(Activation('softmax'))

Моя форма ввода (256,256,2). Я получил эту ошибку: TypeError: Добавленный слой должен быть экземпляром класса Layer. Найдено: Тензор ("input_1: 0", shape = (None, 256, 256, 2), dtype = float32). Что я сделал не так?

Answer 1

Ввод не возвращает слой https://github.com/fchollet/keras/blob/master/keras/engine/topology.py#L1142

Это оболочка InputLayer. Поскольку вы используете Sequential, просто пропустите его и используйте параметр input_shape:

model.add(Flatten(input_shape = your_input_shape)