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

时间:2018-06-30 16:28:48

标签: android tensorflow keras tensorflow-lite

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

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

但是,当我运行此代码时,出现错误消息:

  

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

如何解决这个问题?

3 个答案:

答案 0 :(得分:1)

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

答案 1 :(得分:0)

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

答案 2 :(得分:-1)

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

上提交问题也将有所帮助