java读取pb文件_tf导出pb文件，以及如何使用pb文件

先罗列出来代码，有时间再解释

from tensorflow.python.framework import graph_util

import tensorflow as tf

def export_model(input_checkpoint, output_graph):

#这个可以加载saver的模型

saver = tf.train.import_meta_graph(input_checkpoint + '.meta', clear_devices=True)

graph = tf.get_default_graph() # 获得默认的图

input_graph_def = graph.as_graph_def() # 返回一个序列化的图代表当前的图

with tf.Session() as sess:

sess.run(tf.global_variables_initializer())

saver.restore(sess, input_checkpoint)

output_graph_def = graph_util.convert_variables_to_constants( # 模型持久化，将变量值固定

sess=sess,

input_graph_def=input_graph_def,# 等于:sess.graph_def

output_node_names=['softmax_linear/softmax_linear','Cast_1'])# 如果有多个输出节点，以逗号隔开这个是重点，输入和输出的参数都需要在这里记录

with tf.gfile.GFile(output_graph, "wb") as f: #保