我在这个网站上尝试使用Tensorflow在c ++中保存模型的例子:https://medium.com/jim-fleming/loading-a-tensorflow-graph-with-the-c-api-4caaff88463f#.ji310n4zo
它运作良好.但它不保存变量a和b的值,因为它只保存图形而不保存变量.我试图替换以下行:
tf.train.write_graph(sess.graph_def, 'models/', 'graph.pb', as_text=False)
同
saver.save(sess, 'models/graph', global_step=0)
当然在创建了保护对象之后.它不起作用,它输出:
未找到:FeedInputs:无法找到Feed输出a
我检查了节点加载的节点,它们只是:
_资源
_下沉
在write_graph函数中,然后在C++中加载模型,我加载了以下节点:
_资源
_下沉
保存/ restore_slice_1/shape_and_slice
保存/ restore_slice_1/tensor_name
保存/ restore_slice/shape_and_slice
保存/ restore_slice/tensor_name
保存/保存/ shapes_and_slices
保存/保存/ tensor_names
保存/常量
保存/ restore_slice_1
保存/ restore_slice
b
保存/ Assign_1
面包
B/initial_value
B /分配
一个
保存/分配
保存/ RESTORE_ALL
保存/保存
保存/ control_dependency
A /读
C
A/initial_value
A /分配
在里面
张量
甚至saver.save()创建的图形文件比write_graph创建的图形文件小得多165B(1.9KB).