当前位置: 首页 > 面试题库 >

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

魏彦
2023-03-14
问题内容

我是Tensorflow的新手。我将训练后的模型保存为pb文件,我想使用tensorflow
mobile,并且使用TFLITE文件很重要。问题是在谷歌搜索转换器后发现的大多数示例都是在终端或cmd上命令的。能否请您分享一个使用pythonhtml" target="_blank">代码转换为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

请记住,您可以通过以下两种方式转换为tflite:

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



 类似资料:
  • 我想在Tensorflow中将整数量化的tflite模型转换为冻结图(.pb)。我通读并尝试了许多有关StackOverflow的解决方案,但没有一个有效。具体来说,toco不起作用(输出格式不能是TENSORFLOW\u GRAPHDEF)。 我的最终目标是通过tf2onnx获得量化的ONNX模型,但tf2onnx不支持tflite作为输入(只支持保存的_模型、检查点和图形_定义)。但是,在使用

  • (基本)C:\TensorFlow-master>Bazel run--Config=opt\//TensorFlow/contrib/lite/toco:toco--\--input_file=optimized_graph.pb\--output_file=abc.tflite\--input_format=tensorFlow_graphdef\--output_format=tflite\

  • 本文向大家介绍使用python把json文件转换为csv文件,包括了使用python把json文件转换为csv文件的使用技巧和注意事项,需要的朋友参考一下 了解json整体格式 这里有一段json格式的文件,存着全球陆地和海洋的每年异常气温(这里只选了一部分):global_temperature.json 通过python读取后可以看到其实json就是dict类型的数据,description和

  • 问题内容: 如何使用python脚本将.csv文件转换为.dbf文件?我发现这一块的代码,在网上,但我不能肯定它是多么可靠。是否有任何具有此功能的模块? 问题答案: 在网上找不到任何可以读取CSV文件并写入DBF文件的文件,因此您只需调用它并提供2个文件路径即可。对于每个DBF字段,您需要指定类型,大小和(如果相关)小数位数。 一些问题: 哪些软件将使用输出的DBF文件? 没有所谓的“ the”(

  • 问题内容: 我需要帮助来理解使用java将XML文件转换为CSV文件所涉及的步骤。这是XML文件的示例 这是生成的CSV文件。 我当时正在考虑使用DOM解析器读取xml文件。我遇到的问题是,我需要按名称指定要编码的特定元素,但我希望它能够解析它而不这样做。 Java中是否有任何工具或库可以帮助我实现这一目标。 如果我下面有这种格式的XML文件,并且想在与MSgId相同的行中添加InitgPty的值

  • 我需要将文本文件转换为csv并按列组织它。但是,文本文件中的数据是按行排列的,长度为715页。下面是文本文件外观的示例: 这是我的数据的一个例子,但实际上每个人的信息要多得多,而且有成千上万的条目。 基本上,每一行都用“----”分隔。但是,行之间的数据是多行的。例如,下一行是姓名和年龄,下一行是薪水,下一行是薪水,下一行是薪水,下一行是薪水,这意味着新条目的开始。 有没有一种方法可以让我绕过这种