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实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
- Vertx基于EventBus发送接受自定义对象
- Git恢复之前版本的两种方法reset、revert(图文详解)
- Git撤销&回滚操作(git reset 和 get revert)
- IDEA连接mysql又报错!Server returns invalid timezone. Go to tab and set serverTimezone prope的问题
- PIL包中Image模块的convert()函数的具体使用
- MySQL数据库中CAST与CONVERT函数实现类型转换的讲解
- SpringBoot基于HttpMessageConverter实现全局日期格式化
- 使用IDEA和Gradle构建Vertx项目(图文步骤)
- Vert.x运行环境搭建流程图解
相关文章
Python Flask框架实现Proteus仿真Arduino与网页数据交互
这篇文章主要介绍了Python Flask框架实现Proteus仿真Arduino与网页数据交互,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习吧2022-11-11nginx黑名单和django限速,最简单的防恶意请求方法分享
今天小编就为大家分享一篇nginx黑名单和django限速,最简单的防恶意请求方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08Linux下把Python3设为默认Python版本的三种方法
这篇文章主要给大家介绍了关于Linux下把Python3设为默认Python版本的三种方法,在大部分开发过程中,我们需要使用到python3进行测试,于是下载安装python3,近乎是一种刚需,至于设置默认python指向python3可以参考本文2023-09-09
最新评论