要調用訓練好的TensorFlow模型,你需要進行以下步驟:
tf.saved_model.load
函數加載已經保存的模型。例如:import tensorflow as tf
loaded_model = tf.saved_model.load('path/to/saved/model')
loaded_model.signatures
屬性來獲取模型的推理函數。例如:inference_fn = loaded_model.signatures['serving_default']
import numpy as np
input_data = np.array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=np.float32)
output = inference_fn(tf.constant(input_data))
output.numpy()
將其轉換為NumPy數組。如果模型的輸出是一個字典,你可以使用output['output_name'].numpy()
來獲取特定輸出。例如:output_data = output['output_name'].numpy()
完成以上步驟后,你就成功調用了訓練好的TensorFlow模型,并獲得了推理結果。請根據你的具體情況進行相應的調整。