Keras - ввод изображения в LSTM - PullRequest
Новая
       32

Keras - ввод изображения в LSTM

0 голосов
/ 29 мая 2020

У меня есть последовательность изображений X в массив numpy с формой (14, 300, 300, 3), где 14 - временные шаги, 300 - столбцы, 300 - строки и 3 канала (RGB).

И у меня Y - это массив numpy с формой (300, 300), содержащий кластеры

У меня есть последовательная модель со слоем LSTM: 

model_LSTM = Sequential()
model_LSTM.add(LSTM(300, input_shape = (x_train.shape[1], x_train.shape[2]), dropout=0.2, return_sequences=True))
model_LSTM.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model_LSTM.summary()

hist_LSTM = model_LSTM.fit(X, Y, batch_size=5, epochs=6, validation_data=(x_test,y_test))

Но похоже, что модель либо исчерпывает ОЗУ, либо сообщает мне о несоответствии ввода-вывода (Нет, 10, 300) и (Нет, 300, 300)

Мне нужна помощь: /

...