Tensorflow使用python将PB文件转换为TFLITE

  • Z时代
  • 2024-01-10
  • 分类：问答

我是Tensorflow的新手。我将训练后的模型保存为pb文件，我想使用tensorflow

mobile，并且使用TFLITE文件很重要。问题是在谷歌搜索转换器后发现的大多数示例都是在终端或cmd上命令的。能否请您分享一个使用python代码转换为tflite文件的示例？

谢谢

回答：

在此TF示例之后，您可以传递“

–Saved_model_dir”参数以将save_model.pb和变量文件夹导出到某个目录（不存在现有dir），然后再运行retrain.py脚本：

python retrain.py ...... –saved_model_dir /home/..../export

为了将您的模型转换为tflite，您需要使用以下代码行：

convert_saved_model.convert(saved_model_dir='/home/.../export',output_arrays="final_result",output_tflite='/home/.../export/graph.tflite')

注意：您需要导入convert_saved_model：

从tensorflow.contrib.lite.python导入convert_saved_model

，最简单的方法是使用变量导出exported_model.pb，以防您避免使用Bazel等构建工具。

以上是 Tensorflow使用python将PB文件转换为TFLITE 的全部内容， 来源链接： utcz.com/qa/404421.html

回到顶部