如何在TensorFlow中使用預先訓練的模型

與Caffe一樣，互聯網上也有一些預先訓練好的TensorFlow模型。然而，與Caffe不同，它的模型總是伴隨着一個指示網絡結構和圖層名稱的文件，TensorFlow依靠Python代碼來定義神經網絡。同時，對於TensorFlow中的變量，由於variable_scope的存在，其名稱可能非常長。這使得通過名稱找到特定變量非常困難。所以，如果我從互聯網上下載了一個預先訓練好的TensorFlow模型而沒有網絡結構的python代碼，那我該如何將它用於我自己的項目？如何在TensorFlow中使用預先訓練的模型

來源

2017-07-18 C. Wang

Tensorflow模型可以保存爲protobuf格式。這種格式對於存儲訓練過的模型非常有用，圖形用權重保存爲常量，而不是變量。您可以加載它並通過這些數據初始化您自己的模型並對其進行微調。

https://www.tensorflow.org/api_docs/python/tf/GraphDef

來源

2017-07-18 19:58:06

如何在TensorFlow中使用預先訓練的模型

回答

相關問題