python - 无法运行 EfficientNet 模型来训练 4 类

Question

我将用 4 个带有图像的类来训练我的模型。

这是我预先训练的代码看起来像

conv_base = EfficientNetB0(weights="imagenet", include_top=False,
                                               input_shape=(224, 224, 3))

我用这样的代码训练了我的模式：

eff_history = model.fit(
                        train_generator,
                        validation_data=validation_generator,
                        steps_per_epoch=486 // 20,
                        epochs=10)

但是出现了如下问题：

有什么帮助吗？

Answer 1

每个样本都有三个标签，但是模型的最后一层输出 2 个类，因此您获得的 logit 数是 2 而不是 3。您需要更改模型以获得 3 维输出。