溫馨提示×

TensorFlow中怎么轉(zhuǎn)換模型

小億
94
2024-05-10 18:50:51

要轉(zhuǎn)換TensorFlow模型，可以使用TensorFlow提供的tf.saved_model API。

下面是一個示例代碼，將一個Keras模型轉(zhuǎn)換為SavedModel格式：

import tensorflow as tf

# 加載Keras模型
model = tf.keras.models.load_model('model.h5')

# 轉(zhuǎn)換為SavedModel格式
tf.saved_model.save(model, 'saved_model')

要轉(zhuǎn)換為TensorFlow Lite模型，可以使用tf.lite.TFLiteConverter類，示例如下：

# 轉(zhuǎn)換為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模型轉(zhuǎn)換為SavedModel格式或TensorFlow Lite格式。

0 踩