要调用训练好的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模型,并获得了推理结果。请根据你的具体情况进行相应的调整。