读取pb文件恢复模型

恢复pb文件模型

打开文件：
with tf.gfile.FastGFile(os.path.join(MODEL_DIR, MODEL_FILE), 'rb') as f:
或
with open(os.path.join(MODEL_DIR, MODEL_FILE), 'rb') as f:

graph = tf.Graph()

with tf.gfile.FastGFile(os.path.join(MODEL_DIR, MODEL_FILE), 'rb') as f:
	graph_def = tf.GraphDef()               # 生成图
    graph_def.ParseFromString(f.read())     # 图加载模型
    
    with graph.as_default():
    	return_tensor = tf.import_graph_def(graph_def,return_elements=[TENSOR_A,TENSOR_B])                        # 恢复tensor