0

我在https://www.tensorflow.org/versions/master/tutorials/image_retraining关注本教程

在我经历了一些障碍之后,我运行了 retrain.py。然后我意识到新模型只能对我添加的新类别的图像进行分类。我想知道我在这里做错了什么。我认为除了预先训练的 1000 个类之外,新模型还应该处理新添加的类别。任何输入/评论表示赞赏。谢谢。

4

1 回答 1

0

天真的答案是:不,你不应该。因为再训练背后的想法是迁移学习

根据维基百科

迁移学习侧重于存储在解决一个问题时获得的知识,并将其应用于另一个不同但相关的问题。

所以源标签和目标标签应该不同。

但当然,您可以通过组合源数据集和目标数据集来做您想做的事情,并通过提供组合数据集来运行模型。在您的情况下,培训将花费太多时间,因为初始数据集并不小。

于 2017-12-08T12:58:11.280 回答