ValueError в сверточной нейронной сети

Question

Я получаю сообщение об ошибке: «Ожидается, что conv2d_49_input будет иметь 4 измерения, но получил массив с формой (156, 156, 2)», ну, если я попытаюсь изменить форму ввода на (-1, 156, 156 , 2) или (BATCH_SIZE, 156, 156, 2) он сообщает мне, что входы 5 измерений не ожидаются (None, BATCH_SIZE, 156, 156, 2).

inputShape = X[0].shape # (156, 156, 2)
BATCH_SIZE=32


model = Sequential()
model.add(Conv2D(64, 
                 kernel_size=(3,3), 
                 input_shape=(156,156,2), 
                 activation= 'relu',
                 data_format='channels_last'))
...
...
model.fit(X, y, batch_size=BATCH_SIZE, validation_split=0.1, epochs=10)

Попытка изменить размер массивов np в X на что-то вроде (-1, 156, 156, 2) или (BATCH_SIZE, 156, 156, 2) на самом деле не сработало для меня, что совет, который я видел на этом сайте.

Answer 1

Попробуйте это

import numpy as np

BATCH_SIZE=32
Batch = [X[i] for i in range(BATCH_SIZE)]
input = np.stack(Batch)
print(input.shape)