获取tensorflow lite模型指定中间层的输出

mobilenet v2为例子,我已经训练好了模型,有tflite格式和onnx格式两种模型的权重文件,我想获取tflite模型在推理阶段neck部分的输出。

  1. 查看onnx模型得到neck最后一层位置

使用netron查看onnx模型结构


从name中能知道Reshapeneck的最后一层

  1. 得到neck最后一层的outputtflite模型中的编号

使用netron查看tflite模型结构找到Reshape的输出的编号

  1. 在推理脚本中增加保存中间层输出

其他说明:

偶尔同样的操作就会报错ValueError: Tensor data is null. Run allocate_tensors() first,原因有说是tf的版本的原因,但我不更改版本的情况下有时也出现这个问题,以后再细究吧,保存下当前的tf版本:

相关推荐
Tutankaaa13 小时前
从被动接受到主动挑战:知识竞赛如何重塑学习价值
人工智能·经验分享·笔记·学习
m0_5145205713 小时前
MySQL索引优化后性能没提升_通过EXPLAIN查看索引命中率
jvm·数据库·python
H Journey13 小时前
Python 国内pip install 安装缓慢
python·pip·install 加速
Jmayday14 小时前
机器学习基本理论
人工智能·机器学习
ZhengEnCi14 小时前
01b-上下文向量与信息瓶颈
人工智能
王_teacher14 小时前
机器学习 矩阵求导 完整公式+严谨推导
人工智能·线性代数·考研·机器学习·矩阵·线性回归
码以致用14 小时前
DeerFlow Memory架构
人工智能·ai·架构·agent
ting945200014 小时前
从零构建大模型实战:数据处理与 GPT-2 完整实现
人工智能
学点程序14 小时前
Manifest:帮个人 AI Agent 降低模型成本的开源路由器
人工智能·开源
可观测性用观测云14 小时前
观测云 x AI Agent:运维智能化的范式跃迁实践
人工智能