首页 文章

检查目标时的Keras ValueError

提问于
浏览
1

我正在尝试在keras中 Build 一个模型 . 我几乎遵循了一个教程,但是我收到的错误是:

ValueError:检查目标时出错:期望activation_5具有形状(无,1)但是具有形状的数组(16,13)

我的代码如下:

model = Sequential()
model.add(Conv2D(32, (3, 3), input_shape=input_shape))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Conv2D(32, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Conv2D(64, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Flatten())
model.add(Dense(64))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(1))
model.add(Activation('sigmoid'))

model.compile(loss='binary_crossentropy',
              optimizer='rmsprop',
              metrics=['accuracy'])


batch_size = 16
epochs = 50
number_training_data = 999
number_validation_data = 100

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

test_datagen = ImageDataGenerator(rescale=1. / 255)

train_generator = train_datagen.flow_from_directory(
        'data/train',  # this is the target directory
        target_size=(200, 200),
        batch_size=batch_size,
        class_mode='categorical')

validation_generator = test_datagen.flow_from_directory(
        'data/validation',
        target_size=(200, 200),
        batch_size=batch_size,
        class_mode='categorical')

model.fit_generator(
    train_generator,
    steps_per_epoch=number_training_data // batch_size,
    epochs=epochs,
    validation_data=validation_generator,
    validation_steps=number_validation_data // batch_size)

我有的数据集有13个类,因此错误消息中数组的形状对应于批量大小和类的数量 . 知道我为什么会收到这个错误吗?

1 回答

  • 2

    您的模型配置为执行二进制分类,而不是具有13个类的多类分类 . 要做到这一点,你应该改变:

    • 最后密集的单位数为13,类数 .

    • 输出到softmax的激活 .

    • 分类交叉熵的损失( categorical_crossentropy ) .

相关问题