我正在尝试将经过训练的 Keras 模型从.pb文件转换为.tflite文件：
tflite_model = tf.contrib.lite.toco_convert(frozen_graph_def, input_tensors, output_tensors)
但是，当我运行此代码时，会出现一条错误消息：
AttributeError：模块“tensorflow.contrib.lite”没有属性“toco_convert”
如何解决这个问题呢？
问问题
852 次
3 回答
1
如果您在 Windows 上，它将无法工作，因为 toco 库不支持 Windows。所以最好切换Linux再试一次。
于 2018-11-16T06:52:17.473 回答
0
您使用的是最新版本的 tensorflow 吗？这也可以在安装 tf-nightly 时解决。或者使用 bazel 从源代码编译 tensorflow。您也可以改用：tf.contrib.lite.TocoConverter.from_frozen_graph(final_model_path, input_arrays, output_arrays)
于 2018-08-29T08:18:21.647 回答
-1
请提供您尝试转换的张量流图以重现确切的错误。在 GitHub https://github.com/tensorflow/tensorflow/issues上提交问题也会有所帮助
于 2018-07-06T17:48:49.577 回答