我是深度学习领域的新手,我正努力了解使用keras和数据增强包(ImageDataGenerator)实现自己的VGG16时做错了什么。特别是,我不明白为什么会收到此错误:
检查目标时出错:预期预测的形状为(1000,),但数组的形状为(1,)
我没有传递有关y值的任何信息。确实,我希望keras本身能够处理预测形状,因为我正在使用.flow_from_directory方法。
请查看下面的代码:
train_datagen = ImageDataGenerator(
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
train_generator = train_datagen.flow_from_directory(
train_dir,
target_size=(224, 224),
batch_size=32,
class_mode='binary',
classes = ['NORMAL','PNEUMONIA'])
from keras.applications.vgg16 import VGG16
model = VGG16()
model.compile(optimizer='rmsprop',
loss='binary_crossentropy',
metrics=['accuracy'])
model.fit_generator(
train_generator,
use_multiprocessing=True)
有人可以帮助我吗?我哪里做错了?预先感谢!