如何使用TensorFlow2生成frozen_inference_graphe.pb和.pbtxt文件

pip install tensorflow==2.0

import tensorflow as tf

model = tf.keras.models.load_model('path_to_your_model.h5')

from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2

# 将模型转换为frozen graph
frozen_graph = convert_variables_to_constants_v2(model, lower_control_flow=False)

# 保存frozen graph
tf.io.write_graph(frozen_graph.graph_def, '.', 'frozen_inference_graph.pb', as_text=False)

from tensorflow.python.platform import gfile

# 加载frozen graph
with tf.compat.v1.Session() as sess:
    with gfile.FastGFile('frozen_inference_graph.pb', 'rb') as f:
        graph_def = tf.compat.v1.GraphDef()
        graph_def.ParseFromString(f.read())
        tf.import_graph_def(graph_def, name='')

    # 保存.pbtxt文件
    tf.io.write_graph(sess.graph, '.', 'frozen_inference_graph.pbtxt', as_text=True)