Loading [MathJax]/jax/output/CommonHTML/config.js
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >先将tensorflow2.4cpkt转为.pb模型。再将.pb模型转为.tflite时报错?

先将tensorflow2.4cpkt转为.pb模型。再将.pb模型转为.tflite时报错?

提问于 2022-12-28 02:25:50
回答 0关注 0查看 138

先将tensorflow2.4cpkt转为.pb模型。再将.pb模型转为.tflite时报错

File "D:\Program Files\Python\Python37\lib\site-packages\tensorflow\lite\python\convert.py", line 216, in toco_convert_protos

raise ConverterError(str(e))

tensorflow.lite.python.convert.ConverterError: <unknown>:0: error: loc("bidirectional_rnn/bw/bw/while"): 'tf.WhileRegion' op body result type tensor<1x1x256xf32> is incompatible with result type tensor<0x1x256xf32> at index 7

下面是我.pb转tflite的代码

代码语言:javascript
运行
AI代码解释
复制
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()



# 把pb文件路径改成自己的pb文件路径即可
path = "../model/HTOCR.pb"

# 如果是不知道自己的模型的输入输出节点,建议用tensorboard做可视化查看计算图,计算图里有输入输出的节点名称
inputs = ["Placeholder","Placeholder_4","is_train"]
outputs = ["CTCGreedyDecoder","transpose"]

input_tensor_shape = {'Placeholder': [1, 616, 32],'Placeholder_4':[1],'is_train':[1]}
# 转换pb模型到tflite模型
converter = tf.lite.TFLiteConverter.from_frozen_graph(path, inputs, outputs,input_tensor_shape)
# converter.post_training_quantize = True

tflite_model = converter.convert()
open("../model/HTOCR.tflite", "wb").write(tflite_model)

回答

成为首答用户。去 写回答
相关文章
将训练好的 mmdetection 模型转为 tensorrt 模型
mmdetection 是商汤科技(2018 COCO 目标检测挑战赛冠军)和香港中文大学开源的基于Pytorch实现的深度学习目标检测工具箱,性能强大,运算效率高,配置化编程,比较容易训练、测试。但pytorch模型不易于部署,运算速度还有进一步提升的空间,当前比较有效的方法是将模型转换为行为相同的tensorrt模型,本文记录转换流程。 任务思路 转换mmdetection 的 pytorch模型到tensorrt模型有多种方法,本文使用 mmdetection-to-tensorrt 库作为核
为为为什么
2022/08/05
1.2K0
深度学习算法优化系列五 | 使用TensorFlow-Lite对LeNet进行训练后量化
在深度学习算法优化系列三 | Google CVPR2018 int8量化算法 这篇推文中已经详细介绍了Google提出的Min-Max量化方式,关于原理这一小节就不再赘述了,感兴趣的去看一下那篇推文即可。今天主要是利用tflite来跑一下这个量化算法,量化一个最简单的LeNet-5模型来说明一下量化的有效性。tflite全称为TensorFlow Lite,是一种用于设备端推断的开源深度学习框架。中文官方地址我放附录了,我们理解为这个框架可以把我们用tensorflow训练出来的模型转换到移动端进行部署即可,在这个转换过程中就可以自动调用算法执行模型剪枝,模型量化了。由于我并不熟悉将tflite模型放到Android端进行测试的过程,所以我将tflite模型直接在PC上进行了测试(包括精度,速度,模型大小)。
BBuf
2020/01/21
1.7K0
深度学习算法优化系列五 | 使用TensorFlow-Lite对LeNet进行训练后量化
Caffe2 - (七)Caffemodel 转换为 Caffe2 pb 模型
Caffe2 提供了将 caffemodel 转换为 caffe2 模型的工具——caffe_translator.py.
AIHGF
2019/02/18
1.3K0
Map<String,Object>转为JSONObject ,JSONObject 转为JsonObject,JsonObject转为JsonElement
package com.tianjian.property.jieshuns.service;
全栈程序员站长
2022/08/28
3.2K0
从Tensorflow模型文件中解析并显示网络结构图(pb模型篇)
Tensorflow官方提供的Tensorboard可以可视化神经网络结构图,但是说实话,我几乎从来不用。主要是因为Tensorboard中查看到的图结构太混乱了,包含了网络中所有的计算节点(读取数据节点、网络节点、loss计算节点等等)。更可怕的是,如果一个计算节点是由多个基础计算(如加减乘除等)构成,那么在Tensorboard中会将基础计算节点显示而不是作为一个整体显示(典型的如Squeeze计算节点)。最近为了排查网络结构BUG花费一周时间,因此,狠下心来决定自己写一个工具,将Tensorflow中的图以最简单的方式显示最关键的网络结构。
superhua
2018/11/07
11K1
从Tensorflow模型文件中解析并显示网络结构图(pb模型篇)
[TensorFlow深度学习入门]实战八·简便方法实现TensorFlow模型参数保存与加载(pb方式)
在上篇博文中,我们探索了TensorFlow模型参数保存与加载实现方法采用的是保存ckpt的方式。这篇博文我们会使用保存为pd格式文件来实现。 首先,我会在上篇博文基础上,实现由ckpt文件如何转换为pb文件,再去探索如何在训练时直接保存pb文件,最后是如何利用pb文件复现网络与参数完成应用预测功能。
小宋是呢
2019/06/27
2.2K0
Tensorflow中float32模型强制转为float16半浮点模型
在Tensorflow框架训练完成后,部署模型时希望对模型进行压缩。一种方案是前面文字介绍的方法《【Ubuntu】Tensorflow对训练后的模型做8位(uint8)量化转换》。另一种方法是半浮点量化,今天我们主要介绍如何通过修改Tensorflow的pb文件中的计算节点和常量(const),将float32数据类型的模型大小压缩减半为float16数据类型的模型。
superhua
2020/12/10
3.1K0
Power Designer逆向工程导入Oracle表,转为模型加注释
1.打开PowerDesigner ——文件——Reverse Engineer——DataBase 2.选择所要连接数据库版本,此处使用的是ora1.打开PowerDesigner ——文件——Reverse Engineer——DataBase 2.选择所要连接数据库版本,此处使用的是oracle version 11g。 3.点击红色区域,选择数据源 4.选择modify 5.在此填写你的数据库名称、连接地址、用户名。确定 6.选择你新建立的连接数据库 7.填写需要转换为模型的数据库的用
架构师专栏
2018/07/06
7220
String转为long 类型报错原因:要转为long必须是int、double、float型[通俗易懂]
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/128601.html原文链接:https://javaforall.cn
全栈程序员站长
2022/07/28
2.4K0
Tensorflow将模型导出为一个文件及接口设置
在上一篇文章中《Tensorflow加载预训练模型和保存模型》,我们学习到如何使用预训练的模型。但注意到,在上一篇文章中使用预训练模型,必须至少的要4个文件:
superhua
2018/08/28
1.9K0
PB协议与JSON之间的对比(解惑PB的应用场景)
下个月我要做一个新项目,会用到Java、C++、Python,而版本迭代我想用PB协议,因为我并不知道JSON该如何做版本迭代。
看、未来
2021/09/18
2.2K0
Pb协议的接口测试
Protocol Buffers 是谷歌开源的序列化与反序列化框架。它与语言无关、平台无关、具有可扩展的机制。用于序列化结构化数据,此工具对标 XML ,支持自动编码,解码。比 XML 性能好,且数据易于解析。更多有关工具的介绍可参考官网。
霍格沃兹测试开发
2022/04/29
1.1K0
Protocol Buffers - PB协议概述
� ProtocolBuffers-Google'sdatainterchangeformat
张云飞Vir
2020/03/16
3.3K0
点击加载更多

相似问题

如何将EAST模型训练生成的ckpt文件转换成pb模型文件?

0518

离线语音无法转为文本显示?

0316

xgboost模型报错?

049

域名注册时写了企业,可以转为个人的吗?

41.6K

Noahmp模型运行报错?

018
相关问答用户
擅长3个领域
高级数据分析师擅长5个领域
萃橙科技 | 合伙人擅长4个领域
擅长4个领域
添加站长 进交流群

领取专属 10元无门槛券

AI混元助手 在线答疑

扫码加入开发者社群
关注 腾讯云开发者公众号

洞察 腾讯核心技术

剖析业界实践案例

扫码关注腾讯云开发者公众号
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档