Tensorflow分類圖像訓練

Question

有沒有什麼辦法可以在tensorflow的現有的1000模型中添加一個新的類？或者是否可以通過將更多圖像添加到現有類別來微調模型

Answer 1

請參閱TensorFlow上的教程How to Retrain Inception's Final Layer for New Categories。

從教程：

現代物體識別模型有上百萬的參數，可能需要數週來全面培訓。轉移學習是一種技巧，通過爲ImageNet等一系列類別提供完全培訓的模型，並從現有的新班級權重中重新訓練，從而縮短了大量工作量。在這個例子中，我們將從頭開始重新訓練最後一層，同時保留所有其他層。有關該方法的更多信息，請參見this paper on Decaf。