如何正确导出 TensorFlow LSTM 模型以用于 Android 部署
本文详解 tensorflow 中 lstm 模型导出失败的常见原因，并提供现代、简洁、兼容性强的替代方案——直接使用 keras 的 `.h5` 或 savedmodel 格式保存与加载，避免冻结图（freeze_graph）等过时且易错的流程。
你遇到的 freeze_graph.freeze_graph(...) 报错，根本原因在于：该函数已在 TensorFlow 2.x 中被正式弃用，且其调用逻辑高度依赖 TF 1.x 的会话（Session）、图（Graph）和检查点（Checkpoint）手动管理机制。而你的代码混合了 TF 2.x 风格（如 tf.keras.Sequential、tf.train.Checkpoint）与 TF 1.x 工具链（freeze_graph），导致节点名不匹配、图结构缺失、检查点路径错误（如 'har_model.chkp-1' 实际应为 'har_model.chkp-XXXX' 或索引文件 checkpoint）等一系列不可靠行为——这正是截图中报错（如 NotFoundError: Key lstm_1/kernel not found in checkpoint 或 Cannot parse file ...）的典型根源。
✅ 推荐方案：使用 Keras 原生保存格式（推荐 .h5 或 SavedModel）
Keras 模型（包括含 LSTM 的序列模型）支持开箱即用的完整序列化，自动保存网络结构、权重、优化器状态（可选）及自定义层配置，无需手动处理图或冻结操作：
# ✅ 正确保存（TF 2.x 推荐方式）
har_model.save("har_model.h5") # 保存为 HDF5 格式（轻量、跨平台）
# 或更推荐（尤其面向移动端部署）：
har_model.save("har_model_savedmodel", save_format="tf") # 保存为 SavedModel 目录
? 为什么 SavedModel 更适合 Android？
TensorFlow Lite（TFLite）是 Android 上部署 TF 模型的标准方案，它原生支持从 SavedModel 直接转换：
# 将 SavedModel 转换为 TFLite 模型（适配 Android）
import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_saved_model("har_model_savedmodel")
converter.target_spec.supported_ops = [
tf.lite.OpsSet.TFLITE_BUILTINS,
tf.lite.OpsSet.SELECT_TF_OPS # 若含自定义/高级 LSTM 操作（如 stateful LSTM）
]
tflite_model = converter.convert()
# 保存为 .tflite 文件
with open("har_model.tflite", "wb") as f:
f.write(tflite_model)
⚠️ 关键注意事项：
Looka
AI辅助Logo和品牌设计工具
下载
- 避免混用 TF 1.x 和 TF 2.x API：freeze_graph 属于 TF 1.x 工具集，已从 tensorflow.python.tools 中移除（TF ≥ 2.9+）。强行使用需降级 TF 版本，但将牺牲 Eager Execution、Keras 集成等核心优势。
- LSTM 输入兼容性：Android 端 TFLite 推理要求输入 shape 明确。若模型含动态 batch size（如 None），建议在保存前使用 model.build(input_shape=(1, timesteps, features)) 固定输入规格。
- 检查点路径错误：你代码中 path+'models/har_model.chkp-1' 是无效路径——tf.train.Checkpoint.save() 生成的是带编号的检查点文件（如 har_model.chkp-1.index, .data-00000-of-00001），而 freeze_graph 需要的是不含后缀的前缀路径（如 path+'models/har_model.chkp'），且必须配合正确的 checkpoint 文本文件。此过程极易出错，不推荐手动维护。
✅ 验证加载与推理（确保一致性）：
# 加载并验证
loaded_model = tf.keras.models.load_model("har_model.h5")
print(loaded_model.summary())
# 示例预测（确保输入 shape 匹配训练时设定）
import numpy as np
test_input = np.random.random((1, 100, 6)).astype(np.float32) # batch=1, timesteps=100, features=6
pred = loaded_model.predict(test_input)
print("Prediction shape:", pred.shape)
? 总结：放弃 freeze_graph，拥抱 Keras 原生保存 —— 它更简洁、更鲁棒、更符合 TensorFlow 2.x 最佳实践，且与 TFLite 工具链无缝衔接，是 Android 部署 LSTM 模型的首选路径。