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