要轉換TensorFlow模型,可以使用TensorFlow提供的tf.saved_model API。
下面是一個示例代碼,將一個Keras模型轉換為SavedModel格式:
import tensorflow as tf
# 加載Keras模型
model = tf.keras.models.load_model('model.h5')
# 轉換為SavedModel格式
tf.saved_model.save(model, 'saved_model')
要轉換為TensorFlow Lite模型,可以使用tf.lite.TFLiteConverter
類,示例如下:
# 轉換為TFLite格式
converter = tf.lite.TFLiteConverter.from_saved_model('saved_model')
tflite_model = converter.convert()
# 保存TFLite模型
with open('model.tflite', 'wb') as f:
f.write(tflite_model)
通過以上步驟,您就可以將TensorFlow模型轉換為SavedModel格式或TensorFlow Lite格式。