Tensorflow随附的Android示例为InceptionV3下载一个protobuf文件,其中包含图形和模型中的值。在文档中,我只能找到如何序列化图表(tf.Graph.as_graph_def
)或将变量值保存为tf.train.Saver
。如该示例所示,如何将所有内容保存到单个文件中?
我对此主题回答了一个类似的问题:是否存在一个示例,该示例如何生成包含受过训练的Tensorflow图的protobuf文件?
其基本思想是利用tf.import_graph_def()
替换与常量原来的(培训)图中的变量,然后将得到的写出来GraphDef
使用tf.Graph.as_graph_def()
。