如何在TensorFlow

时间:2017-07-18 19:08:23

标签: python tensorflow

与Caffe一样,互联网上也有一些预先训练过的TensorFlow模型。但是,与Caffe不同,Caffe的模型总是附带一个.prototxt文件,表明网络结构和图层名称,TensorFlow依靠Python代码来定义神经网络。同时,对于TensorFlow中的变量,由于variable_scope的存在,其名称可能很长。这使得通过名称找到特定变量非常困难。因此,如果我从互联网上下载了没有网络结构python代码的预先训练的TensorFlow模型,我该如何将它用于我自己的项目?

1 个答案:

答案 0 :(得分:0)

Tensorflow模型可以以protobuf格式保存。这种格式对于存储训练模型非常有用,图形以权重作为常数保存,而不是变量。您可以加载它并通过此数据初始化您自己的模型并进行微调。

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