您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

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

Tensorflow使用python将PB文件转换为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等构建工具。

python 2022/1/1 18:38:32 有246人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶