获取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版本:

相关推荐
一步一个脚印ッ10 分钟前
AI大模型推理:本质是经验推理而非逻辑推理
人工智能
智慧物业老杨11 分钟前
老旧小区团购模式下的数智化工单体系搭建与实践——基于保利项目600+工单的技术落地解析
java·人工智能·后端·科技·struts
weixin_5118404711 分钟前
2026年5月7日 AI发展对卫星通讯的影响及太空算力中心建设与发展深度研究
人工智能·市场分析
莱歌数字12 分钟前
理论+仿真+实验:方案设计的“三驾马车”,如何帮您降低试错成本?
人工智能·科技·电脑·制造·散热
LNN202212 分钟前
半导体设备 UI 开发工程师:完整工作执行手册
开发语言·python·ui
TG_yunshuguoji14 分钟前
阿里云代理商:企业如何通过 DeepSeek V4 + 阿里云实现 AI 成本大幅降低?
人工智能·阿里云·ai智能体·deepseek v4
卡次卡次117 分钟前
14.1: 总结本章 Python 高性能并发:多线程+多进程核心知识点+实战指南(面试/开发双适配)
服务器·python·面试
连线Insight18 分钟前
3.5亿月活后,豆包收费是一场冒险吗?
大数据·人工智能
大树8819 分钟前
PUE 1.25红线遇上2300瓦芯片:为什么数据中心开始“算不过来账“了?
人工智能
dayuOK630721 分钟前
不会写文案?我用“看图说话”的方法,10分钟搞定一篇
人工智能·职场和发展·新媒体运营·媒体