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

相关推荐
财经资讯数据_灵砚智能几秒前
基于全球经济类多源新闻的NLP情感分析与数据可视化(日间)2026年6月4日
人工智能·python·ai·信息可视化·自然语言处理·ai编程·灵砚智能
侃谈科技圈1 分钟前
2026免费AI自动抠图工具汇总:全平台+电脑在线全方案,无水印零套路
人工智能
apcipot_rain2 分钟前
计科八股20260604——AI安全、K-means、SVM、nano
人工智能·神经网络·安全·支持向量机·kmeans
Rain5094 分钟前
实战:搭建 AI Code Review 自动化流水线
前端·人工智能·git·ci/cd·自动化·ai编程·代码复审
Promise微笑7 分钟前
洞察无形:红外热像仪应用场景与高性价比之选
人工智能·物联网·算法
用户67573181940258 分钟前
两个Bot不能聊天,我让它们自己建了一条高速公路
python
lulu12165440789 分钟前
2026年-企业级大模型API网关实战指南: 微元算力聚合平台性能优化实测
java·人工智能·spring·性能优化·ai编程
如烟花的信页11 分钟前
易盾滑块逆向分析
javascript·爬虫·python·js逆向
2601_9557819813 分钟前
Windows 环境快速部署 Hermes 智能 Agent,规避环境配置各类坑点
人工智能·本地部署·教程分享·hermes·hermes部署
DO_Community14 分钟前
LawVo 借助 DigitalOcean 扩展 AI 驱动的法律服务
人工智能