expected dense_218_input to have 2 dimensions, but got array with shape (512, 28, 28, 1)

Question

I am trying to augment my MNIST dataset in keras but for some reason its not working. Any help will be appreciated.

Part of the code:

x_train = x_train.r         


        

Answer 1

The dense_218_input should be a numpy 2d-array instead of shape: (512, 28, 28, 1). You can reshape it with numpy.reshape.