恢复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