将.keras模型转换为.tflite

Question

我正在尝试将 .keras 模型转换为 .tflite，因为我需要优化我的手部识别模型。 这是我的可疑台词：

model = tf.keras.models.load_model('model/model_save/hand_tracking_model.keras')
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()

我拥有的软件包版本： Python 3.9.13 张量流2.15.0 tf-nightly 2.17.0.dev20240306 tflite 2.10.0

Answer 1

您可能会更轻松地直接从 PyTorch 进行转换：https://github.com/google-ai-edge/ai-edge-torch