Keras CNN中的输入数组

问题描述

'是什么?'在输入和输出数组中是什么意思? 我要做什么才能使其“无”? (请参考所附图片

这是我获取输入数据的代码

from keras.preprocessing.image import ImageDataGenerator

train_datagen = ImageDataGenerator(rescale = 1./255,shear_range = 0.2,zoom_range = 0.2,horizontal_flip = True)

test_datagen = ImageDataGenerator(rescale = 1./255)

training_set = train_datagen.flow_from_directory(train_path,target_size = (128,128),batch_size = 64,class_mode = 'categorical')

test_set = test_datagen.flow_from_directory(valid_path,class_mode = 'categorical')

enter image description here

解决方法

Tensorflow使用四个通道,其中第一个(由?或None表示)代表您将在训练时通过的batch_size。您可能想知道为什么没有?好吧,因为batch_size是灵活的,并且始终由用户定义,所以Tensorflow使用?或无作为占位符,直到通过批处理大小。其他三个渠道显然是维度