2017-06-15 217 views
0

有没有什么办法可以在tensorflow的现有的1000模型中添加一个新的类?或者是否可以通过将更多图像添加到现有类别来微调模型Tensorflow分类图像训练

+1

谁赞赏这个东西? –

+1

好奇,只想知道它与git有什么关系? – LethalProgrammer

回答

0

请参阅TensorFlow上的教程How to Retrain Inception's Final Layer for New Categories

从教程:

现代物体识别模型有上百万的参数,可能需要数周来全面培训。转移学习是一种技巧,通过为ImageNet等一系列类别提供完全培训的模型,并从现有的新班级权重中重新训练,从而缩短了大量工作量。在这个例子中,我们将从头开始重新训练最后一层,同时保留所有其他层。有关该方法的更多信息,请参见this paper on Decaf

+0

,但是我们可以采用与softmax相同的权重,而不是从头开始训练最终图层,并使用它 –