TimeDistributed 2D CNN ValueError: длина шага должна быть 1, 1 или 3, но было 2 - PullRequest
0 голосов
/ 27 мая 2020

Я работаю над слоями TimeDistributed для своих видеоданных. Мои входные данные имеют размер (28620, 40, 140, 3) -> (28620 кадров, высота = 40, ширина = 140, 3 канала RGB).

model = Sequential()

model.add(TimeDistributed(Conv2D(9, kernel_size=(3,3)), input_shape=(8, 40, 140, 3)))
model.add(TimeDistributed(Flatten()))
model.add(LSTM(1024, activation='relu', return_sequences=False))
model.add(Dense(250))
model.add(Dense(21))

Когда мне подходит следующее:

history = model.fit(input_train, gt_train, batch_size=8, epochs=50)  

Я получаю сообщение об ошибке:

ValueError: strides should be of length 1, 1 or 3 but was 2

Я считаю, что ошибка относится к input_shape первого слоя, но я не могу понять, как поступить с этим сообщением об ошибке.

Я ценю каждый намек или идею, как это исправить.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...