Первый слой модели, который является двумерной сверткой, ожидает (я полагаю) данные двумерного изображения с несколькими каналами. Что-то вроде:
(img_height, img_width, num_channels).
Кроме того, вам нужно иметь другое измерение для «пакета» или «образца». Таким образом, окончательная форма будет выглядеть следующим образом:
(num_samples, img_height, img_width, num_channels).
Итак, вам необходимо изменить свои входные данные, чтобы они были в указанном выше шаблоне, с различнымиизмерения, имеющие указанную мною интерпретацию, основаны на том, что я понял из вашего кода.
Вы можете посмотреть на такие функции, как np.reshape и np.expand_dims, в зависимости от того, как ваши исходные данные отформатированы в первую очередь. .
Вы можете создать некоторые тестовые данные, например, такие:
x_train = np.random.random((4,32,32,3))
# then try out the model prediction
model.predict(x_train)
, которые представляют собой четыре изображения размером 32x32 и 3 канала, для проверки вашей обработки.
You 'Также вам нужно будет создать что-то подходящее для y_train, когда вы будете выполнять фактическую подгонку.
Надеюсь, это поможет.