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

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

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