运行VGG16时出现ValueError:预测形状-Keras

时间:2019-11-08 10:17:55

标签: python tensorflow image-processing keras vgg-net

我是深度学习领域的新手,我正努力了解使用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)

有人可以帮助我吗?我哪里做错了?预先感谢!

0 个答案:

没有答案