如何在使用迁移学习训练 Resnet-50 时解决此错误？答案

【问题标题】：How do I solve this error while training Resnet-50 using transfer learning?如何在使用迁移学习训练 Resnet-50 时解决此错误？
【发布时间】：2020-10-21 19:11:29
【问题描述】：

我正在使用 keras。版本 '2.3.1' 使用迁移学习训练 Resnet-50。

运行以下代码时出现错误：

history = model.fit_generator(generator=train_generator,
                              epochs= 3,
                              validation_data=validation_generator)

我得到的错误是：

Epoch 1/3
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-22-a42d5d67d1c7> in <module>
      1 history = model.fit_generator(generator=train_generator,
      2                               epochs= 3,
----> 3                               validation_data=validation_generator)
      4 
      5 

ValueError: Error when checking target: expected dense_2 to have shape (2,) but got array with shape (9,)

有谁知道我该如何解决这个问题？

【问题讨论】：

标签： python keras deep-learning resnet transfer-learning

【解决方案1】：

这仅仅是因为模型的输出形状与您的训练 y（目标文件）不匹配。提供完整的脚本，例如model_summary 和训练数据的形状(x, y)

【讨论】：

我应该在代码中添加什么？你能用代码解释一下吗？
是的，添加代码以及有关数据集的详细信息。那将不胜感激。