将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'

如何解决这个问题？

Answer 1

如果您使用的是Windows，则因为toco库不支持Windows，因此无法使用。因此，最好切换到Linux再试一次。

Answer 2

您使用的是Tensorflow的最新版本吗？每晚安装tf时也可以解决此问题。或使用bazel从源代码编译张量流。 您也可以改用：tf.contrib.lite.TocoConverter.from_frozen_graph(final_model_path, input_arrays, output_arrays)

Answer 3

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

上提交问题也将有所帮助