android - 将 keras 模型从 pb 文件转换为 tflite 文件

Question

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

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

但是，当我运行此代码时，会出现一条错误消息：

AttributeError：模块“tensorflow.contrib.lite”没有属性“toco_convert”

如何解决这个问题呢？

标签： androidtensorflowkerastensorflow-lite

Answer 1

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