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

相关推荐
阿里云大数据AI技术12 小时前
DeepSeek-V4来啦!PAI已支持一键部署,共同迈向百万上下文普惠时代!
人工智能·agent·deepseek
云飞云共享云桌面12 小时前
精密机械制造工厂研发部门使用SolidWorks和ug,三维设计云桌面如何选择?
大数据·运维·服务器·网络·数据库·人工智能·制造
冬奇Lab12 小时前
一天一个开源项目(第82篇):Harness Engineering —— 从“提示语工程”到“驭缰工程”的范式跃迁
人工智能·开源·资讯
itzixiao12 小时前
L1-051 打折(5分)[java][python]
java·python·算法
千里念行客24012 小时前
锚定AI赛道释放红利:安凯微2026年Q1业绩显成色
大数据·人工智能·科技·安全
隔壁大炮12 小时前
第一章_机器学习概述_06.机器学习_模拟拟合问题
人工智能·机器学习
Raink老师13 小时前
【AI面试临阵磨枪】OpenClaw 与企业级 Harness 区别:个人 vs 生产、隔离强度、可观测性?
人工智能·ai 面试
隔壁大炮13 小时前
第一章_机器学习概述_05.机器学习_特征工程介绍
人工智能·机器学习
HappyAcmen13 小时前
10.常见报错排查与基础调试
开发语言·python
北京软秦科技有限公司13 小时前
IA-Lab驱动碳纤维检测升级:AI 检测报告生成助手如何重塑力学性能报告新范式
人工智能