pb编译生成tflite文件

最新推荐文章于 2022-10-21 16:19:14 发布

kkongyu

最新推荐文章于 2022-10-21 16:19:14 发布

阅读量1.5k

点赞数

分类专栏： tensorflow android 文章标签： tensorflow

本文链接：https://blog.csdn.net/ting_163/article/details/81288981

版权

这篇博客介绍了如何使用TensorFlow将.pb模型文件转换为tflite格式，包括通过冻图生成tflite文件的步骤和通过.ckpt与结构.pb文件的转换方法。同时，博主分享了adb调试过程以及遇到的Bug1及其解决策略，提供了参考资料链接。

摘要由CSDN通过智能技术生成

通过冻图生成tflite文件：

tensorflow-master$> bazel build tensorflow/contrib/lite/toco:toco

tensorflow-master$> bazel-bin/tensorflow/contrib/lite/toco/toco \
	--input_file=/home/tclxa/TCL/tmp/frozen_model.pb  \
	--input_format=TENSORFLOW_GRAPHDEF \
	--output_format=TFLITE \
	--output_file=/home/tclxa/TCL/tmp/mobilenet_quant_v1_224.tflite \
	--inference_type=FLOAT \
	--input_type=FLOAT \
	--input_arrays=input \
	--output_arrays=MobilenetV1/Predictions/Reshape_1 \
	--input_shapes=1,224,224,3

–output_file、–inference_type、–input_type、–input_arrays、–output_arrays、–input_shapes需要根据自己模型说明作相应的修改

通过ckpt和结构pb生成冻图：.ckpt和结构.pb文件

tensorflow-master$> bazel build tensorflow/python/tools:freeze_graph

tensorflow-master$> bazel-bin/tensorflow/python/tools/freeze_graph \
	--input_graph=/home/tclxa/TCL/tmp/mobilenet_v1_224.pb \
	--input