簡體   English   中英

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

[英]How to use pre-trained model in TensorFlow

像Caffe一樣,互聯網上也有一些經過預訓練的TensorFlow模型。 但是,與Caffe的模型始終伴隨着指示網絡結構和層名稱的.prototxt文件不同,TensorFlow依賴於Python代碼來定義神經網絡。 同時,對於TensorFlow中的變量,由於存在variable_scope ,其名稱可能很長。 這使得很難通過名稱查找特定變量。 因此,如果我從互聯網上下載了經過預訓練的TensorFlow模型而沒有網絡結構python代碼,那么如何在我自己的項目中使用它呢?

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

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

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM