要將TensorFlow模型保存為.pb文件,可以通過以下步驟來實現:
import tensorflow as tf
# 加載已經訓練好的模型
model = tf.keras.models.load_model('path_to_your_model.h5')
# 獲取模型的圖結構和權重參數
model_graph = tf.get_default_graph()
from tensorflow.python.framework import graph_io
# 定義輸出.pb文件的路徑
output_pb_path = 'path_to_output_pb_file.pb'
# 將模型的圖結構和權重參數保存為.pb文件
with model_graph.as_default():
sess = tf.keras.backend.get_session()
graph_io.write_graph(sess.graph, './', output_pb_path, as_text=False)
總的來說,通過以上步驟,可以簡單地將訓練好的TensorFlow模型保存為.pb文件,以便后續使用。