Keras dimension mismatch with ImageDataGenerator
问题 I am attempting to 'flow' my data into a neural network with Keras. I am using the .flow_from_directory method and the process is giving me fits. I am using the basic example from the keras documentation (I am using tensorflow): ROWS = 64 COLS = 64 CHANNELS = 3 from keras.preprocessing.image import ImageDataGenerator train_datagen = ImageDataGenerator( rescale=1./255) test_datagen = ImageDataGenerator(rescale=1./255) train_generator = train_datagen.flow_from_directory( 'train', target_size=