Tensorflow 模型转换 .pb convert to .lite实例
作者：cefengxu 发布时间：2023-08-13 05:19:04
标签：Tensorflow,pb,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 |
来源：https://blog.csdn.net/u011292539/article/details/80484444
0
投稿猜你喜欢
- XML文档因为其固有的描述性特性而趋向于变得很罗嗦。其结果是文档会由于被描述的数据增多而变得很长，而这种很大的文档会在需要同其他实体进行交换
- 最近在工作中涉及到判断服务器所在ip反馈程序使用情况的程序主要要求就是，本机或局域网调试程序时，不反馈其域名（localhost）或ip站长
- 获取指定日期月份的第一天，你可以使用DATEADD函数，减去指定日期的月份过去了的天数，即可。 代码如下:CREATE FUNC
图像噪声是指存在于图像数据中的不必要的或多余的干扰信息。在噪声的概念中，通常采用信噪比（Signal-Noise Rate, S- <?php $url='test.php?1=1'; $contents="fjka;fjsa;#page#
- 在CSS中，实现分栏布局有两种方法。第一种方法是使用四种CSS定位选项(absolute 、static、relative和fixed)中的
- OL是有序列表，但给list-style-type:decimal 定义，在IE中却显示的全是1，在火狐、Opera、谷歌、Safari等中
- 由于DOM(文档对象模型)概念的推出,这个API使HTML如虎添翼,但是有些学DHTML的朋友还是有些困挠,只是因为目前的手册的书写不太科学
- Css tranistions允许元素的属性在单位时间内发生平滑的过渡，在阅读完《CSS Transitions Module Level
- 参数Parameters解析响应时间resolveTimeout 数据类型：长整型。简单地说就是程序对目标主机的名字解析解析的一个过程时间。
- 本文介绍了使用xmlhttp处理远程文件数据、或采集文章时，对对方网页编码的处理方法。因为使用ajax的xmlhttp网页编码处理不当很容易
- microtime() 函数返回当前 Unix 时间戳的微秒数。用于检测程序执行时间的函数,也是PHP内置的时间函数之一,在PHP中可以用于
- 最近在做搜索设计时，发现了两个容易纠结的小问题，在这里谈谈自己的一些分析。问题一：提交的关键字是哪个？凡客的这个例子中，搜索建议“时尚斜拉链
- ExecuteReader(),ExecuteNonQuery(),ExecuteScalar(),ExecuteXmlReader()之间
- 具体特征如下： 1、通过模板实现俄文正常。 2、通过后台数据库生成的静态俄文信息，后台显示正常， 前台乱码。 3、英文正常。 和该主题相关的
- 茎叶图from itertools import groupbynums2=[225, 232,232,245,235,245,270,22
- 在现代LOGO设计当中，叶子的形状被视做好的创意。或者说，是一种变革的想法。在网页中他们大多被用于轻量级的解决方案、干净的不抽像的设计。在实
- PDOStatement::debugDumpParamsPDOStatement::debugDumpParams — 打印一条 SQL
- 在我们设计网页的时候，总会遇到一些不愉快的事情，最常见的莫过于在后台添加内容后才发现显示的页面被撑开，导致网页极度不美观。以前大
- 本文实例讲述了php+mysqli使用面向对象方式更新数据库的方法，分享给大家供大家参考。具体实现方法如下：<?php//第一步：创建
