张量流开始重新培训

Question

我在https://www.tensorflow.org/versions/master/tutorials/image_retraining

上关注了本教程

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

Answer 1

天真的答案是：不，你不应该。因为重新培训背后的想法是转移学习。

根据维基百科

转移学习 专注于存储在解决一个问题时获得的知识并将其应用于不同但相关的问题。

因此源标签和目标标签应该不同。

但是当然，您可以通过组合源数据集和目标数据集来执行您想要的操作，并通过提供组合数据集来运行模型。在你的情况下，培训会花费太多时间，因为初始数据集并不是很小。