将 keras 模型从 pb 文件转换为 tflite 文件答案

【问题标题】：Convert keras model from pb file to tflite file将 keras 模型从 pb 文件转换为 tflite 文件
【发布时间】：2018-12-09 12:40:11
【问题描述】：

我正在尝试将经过训练的 Keras 模型从 .pb 文件转换为 .tflite 文件：

tflite_model = tf.contrib.lite.toco_convert(frozen_graph_def, input_tensors, output_tensors)

但是，当我运行此代码时，会出现以下错误：

AttributeError: 模块 'tensorflow.contrib.lite' 没有属性 'toco_convert'

如何解决这个问题？

【问题讨论】：

标签： android tensorflow keras tensorflow-lite

【解决方案1】：

如果您在 windows 上，它将无法工作，因为 toco 库不支持 windows。所以最好切换Linux再试一次。

【讨论】：

【解决方案2】：

您使用的是最新版本的 tensorflow 吗？这也可以在安装 tf-nightly 时解决。或者使用 bazel 从源代码编译 tensorflow。你也可以改用：tf.contrib.lite.TocoConverter.from_frozen_graph(final_model_path, input_arrays, output_arrays)

【讨论】：

【解决方案3】：

请提供您尝试转换的张量流图以重现确切的错误。在 GitHub https://github.com/tensorflow/tensorflow/issues

上提交问题也会有所帮助

【讨论】：