Неправильная размерность, переходящая в сверточный слой в тензорном потоке

Question

У меня 15 изображений размером 28х28. Таким образом, мой images.shape есть (15, 28, 28). Тем не менее, я получаю эту ошибку:

ValueError: Error when checking input: expected conv2d_22_input to have 4 dimensions, but got array with shape (15, 28, 28)

Я пробовал различные методы изменения формы и могу заставить это работать на одном изображении формы (1, 28, 28, 1). Что я делаю для списка изображений?

Answer 1

Вам необходимо добавить размер канала обратно в пакет. Измените значение на (15,28,28,1) или (15,1,28,28) в зависимости от вашей конфигурации (по умолчанию каналы в кератах являются последними). ​​

Answer 2

numpy.expand_dims(images, axis=4) работа для меня