1

我正在学习如何使用带有 TF 后端的 Keras 进行图像识别，所以我仍然不确定我在这里做错了什么。

我正在尝试堆叠 2 个模型，一个是VGG16，另一个是我为学习如何堆叠而制作的随机模型。我想在 5 个类别中对图像进行分类。

问题出在最后一部分，当我运行 fit_generator 时。它不是产生一个有效的元组，而是产生看起来像一个列表的东西。我见过很多人遇到类似的问题，但在他们的情况下，输出为 None，所以我不确定解决方案是否相同。

参数

nb_train_samples = 576
nb_validation_samples = 144
epochs = 30
batch_size = 12
img_width, img_height = 150, 150

发电机

train_datagen = ImageDataGenerator(
    rescale=1./255,
    rotation_range=50,
    width_shift_range=0.3,
    height_shift_range=0.3,
    shear_range=0.4,
    zoom_range=0.4,
    horizontal_flip=True)

test_datagen = ImageDataGenerator(rescale=1. / 255)

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode=None,
    shuffle=False)

validation_generator = test_datagen.flow_from_directory(
    validation_data_dir,
    target_size=(img_width, img_height),
    batch_size=batch_size,
    class_mode=None,
    shuffle=False)

我的模型

input = Input(batch_shape=model.output_shape)
x = Flatten()(input)
x = Dense(256, activation='relu', name="new_block_1")(x)
x = Dropout(0.5)(x)
x = Dense(256, activation='relu', name="new_block_2")(x)
x = Dropout(0.5)(x)
x = Dense(5, activation='softmax', name="new_block_3")(x)
top_model = Model(input,x)

input = Input(shape=(img_width, img_height, 3))
x = model(input)
x = top_model(x)
final_model = Model(input, x)

final_model.compile(optimizer='rmsprop',
              loss='categorical_crossentropy', metrics=['accuracy'])

拟合与错误

final_model.fit_generator(
    train_generator,
    steps_per_epoch=nb_train_samples // batch_size,
    epochs=epochs,
    validation_data=validation_generator,
    validation_steps=nb_validation_samples // batch_size)

ValueError: output of generator should be a tuple `(x, y, sample_weight)` or `(x, y)`. Found: [[[[ 0.89411771  0.89019614  0.87450987]
   [ 0.89411771  0.89019614  0.87450987]
   [ 0.89411771  0.89019614  0.87450987]
   ...,

更新 1：根据@petezurich 的提示，将激活函数从“sigmoid”更改为“softmax”

4

1 回答 1

1

您的模型缺少训练标签。

只需在生成器中设置为class_mode，categorical然后将图像放在每个类的子文件夹中。生成器可以很好地从中派生类标签。

于 2017-06-19T20:16:18.720 回答