Keras ValueError: Input 0 is incompatible with layer conv2d_1: expected ndim=4, found ndim=5

python tensorflow deep-learning keras conv-neural-network
Lucky picture Lucky · Dec 6, 2017 · Viewed 74.3k times · Source

I have checked all the solutions, but still, I am facing the same error. My training images shape is (26721, 32, 32, 1), which I believe it is 4 dimension, but I don't know why error shows it is 5 dimension. 

 model = Sequential()

 model.add(Convolution2D(16, 5, 5, border_mode='same', input_shape= input_shape ))

So this is how I am defining model.fit_generator 

model.fit_generator(train_dataset, train_labels, nb_epoch=epochs, verbose=1,validation_data=(valid_dataset, valid_labels), nb_val_samples=valid_dataset.shape[0],callbacks=model_callbacks)

Answer

Daniel Möller picture Daniel Möller · Dec 6, 2017

The problem is input_shape.

It should actually contain 3 dimensions only. And internally keras will add the batch dimension making it 4.

Since you probably used input_shape with 4 dimensions (batch included), keras is adding the 5th.

You should use input_shape=(32,32,1).

Related questions

TypeError: Unexpected keyword argument passed to optimizer: learning_rate

I am trying to load a Keras model which was trained on an Azure VM (NC promo). But I am getting the following error. TypeError: Unexpected keyword argument passed to optimizer:learning_rate EDIT: Here is the code snippet that …

Read More
Keras, How to get the output of each layer?

I have trained a binary classification model with CNN, and here is my code model = Sequential() model.add(Convolution2D(nb_filters, kernel_size[0], kernel_size[1], border_mode='valid', input_shape=input_shape)) model.add(Activation('relu')) model.add(Convolution2…

Read More
How to import keras from tf.keras in Tensorflow?

import tensorflow as tf import tensorflow from tensorflow import keras from keras.layers import Dense I am getting the below error from keras.layers import Input, Dense Traceback (most recent call last): File "<ipython-input-6-b5da44e251a5>", …

Read More