InvalidArgumentError: вход должен быть 4-мерным [8,6171,4]

Question

У меня сеть 2d сверточная.Мой вход имеет 3 измерения, однако я получаю эту ошибку 4 измерений: ошибка измерения

Как вы видите, мой ввод имеет правильные размеры:

правильныйвходное измерение

вот мой код:

from keras import models
from keras import layers
model = models.Sequential()
model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(8,6171,4)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))

model.add(layers.Flatten())
model.add(layers.Dense(64, activation='relu'))
model.add(layers.Dense(10, activation='softmax'))

Почему тогда он запрашивает 4-D, когда я только построил 3-D входной слой?

Пожалуйста, помогите.Спасибо.

Answer 1

Вам необходимо использовать

new_image = tf.expand_dims(image,0)

, поскольку модель ожидает набор данных вместо отдельных изображений.