【发布时间】:2020-09-13 05:53:47
【问题描述】:
下面的代码便于理解,
#Fitting the CNN to the images
from keras.preprocessing.image import ImageDataGenerator
train_datagen = ImageDataGenerator(
rescale = 1.0 / 255,
shear_range = 0.2,
zoom_range = 0.2,
horizontal_flip = True
)
test_datagen = ImageDataGenerator(rescale= 1.0 / 255)
training_set = train_datagen.flow_from_directory(
'dataset/training_set',
target_size = (64,64),
batch_size = 32,
class_mode = 'binary'
)
test_set = test_datagen.flow_from_directory(
'dataset/test_set',
target_size = (64,64),
batch_size = 32,
class_mode = 'binary'
)
当我使用 sparse / categorical 时。它显示相同的输出,
在输出下方,
Found 214 images belonging to 5 classes.
Found 20 images belonging to 5 classes.
注意:此代码用于训练数据集,但不幸的是它不能训练
所以需要帮助
【问题讨论】:
标签: python keras conv-neural-network