Tensorflow 模型转换 .pb convert to .lite实例

更新时间：2020年02月12日 09:28:14 作者：cefengxu

今天小编就为大家分享一篇Tensorflow 模型转换 .pb convert to .lite实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

2017年底，Tensorflow 推出Lite版本，可实现移动端的快速运行，其中，一个很关键的问题，如何把现有分类模型(.pb) 转换为(.lite)模型呢？

其实，步骤如下

1- 进入 Tensorflow 源码文件夹（以便bazel可以无需配置找打相应路径）

2- 转换可执行文件

bazel run --config=opt \
 //tensorflow/contrib/lite/toco:toco -- \
 --input_file=/tmp/mobilenet_v1_1.0_224_frozen.pb \  # 待转换模型路径
 --output_file=/tmp/tflite_model2.tflite \      # 目标模型路径
 --input_format=TENSORFLOW_GRAPHDEF \
 --output_format=TFLITE \
 --input_shape=1,224,224,3 \              # 输入图像宽高
 --input_array=input \                 # 输入节点名称<参考1>
 --output_array=MobilenetV1/Predictions/Reshape_1 \  # 输出节点名称<参考1>
 --inference_type=FLOAT \               # 图像数据类型
 --input_data_type=FLOAT

参考1： 如果输入输出节点不知道：进入python，import tf ，并通过如下命令（ tf.GraphDef() ）查找:

>>> import tensorflow as tf
>>> gf = tf.GraphDef()
>>> gf.ParseFromString(open('/your/path/to/graphname.pb','rb').read())
>>> for n in gf.node:
>>>   print ( n.name +' ===> '+n.op )

参转换后，在android手机上运行，在不考虑精度的前提下，不同实现方式得到的结果如下:

模型	type	model size	speed
mobilenet-v1-224	.pb	17 mb	106 ms
mobilenet-v1-224	.lite	16.4 mb	72 ms

以上这篇Tensorflow 模型转换 .pb convert to .lite实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python基于内置库pytesseract实现图片验证码识别功能
这篇文章主要介绍了Python基于内置库pytesseract实现图片验证码识别功能,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-02-02
关于pycharm python3.7成功安装dlib库的问题
这篇文章主要介绍了pycharm python3.7成功安装dlib库的解决方法，本文分步骤给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2021-12-12
python实现滑雪游戏
这篇文章主要为大家详细介绍了python实现滑雪游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2020-02-02
pygame 键盘事件的实践
本文主要介绍了pygame 键盘事件，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-11-11
解决Pycharm的项目目录突然消失的问题
今天小编就为大家分享一篇解决Pycharm的项目目录突然消失的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
python3 如何解压缩.gz文件
这篇文章主要介绍了python3 如何解压缩.gz文件，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-09-09
Python基于回溯法子集树模板实现8皇后问题
这篇文章主要介绍了Python基于回溯法子集树模板实现8皇后问题,简单说明了8皇后问题的原理并结合实例形式分析了Python回溯法子集树模板解决8皇后问题的具体实现技巧,需要的朋友可以参考下
2017-09-09
python3如何去除字符串中的特殊字符
这篇文章主要介绍了python3如何去除字符串中的特殊字符,在平时处理字符串的时候,经常会遇到字符串中夹杂着我们不希望看到的特殊字符,那么如何处理这些特殊字符呢,今天就跟着小编来看看吧
2023-04-04
Python 如何将 matplotlib 图表集成进到PDF 中
这篇文章主要介绍了Python 如何将 matplotlib 图表集成进到PDF 中，文章介绍内容详细，具有一定的参考价值，需要的小伙伴可以参考一下，希望对你的学习有所帮助
2022-03-03
Android应用开发中Action bar编写的入门教程
这篇文章主要介绍了Android应用开发中ActionBar编写的入门教程,Action Bar可以实现的功能很多,比如导航菜单和标签页切换等,需要的朋友可以参考下
2016-02-02