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

相关推荐
字节跳动视频云技术团队26 分钟前
豆包视频通话升级,火山引擎多模态传输系统提供技术支撑
人工智能·音视频开发
searchforAI30 分钟前
AI自动总结YouTube视频:英文内容一键总结、智能时间戳、视频重点快速看
人工智能·ai·音视频
cxk180827231 分钟前
2026 年9月 AI 营销机构选型方法论:基于 GEO 精采信模式的三维评估体系
大数据·人工智能·科技·ai·geo·昆明geo
威视锐科技32 分钟前
AI-RAN:下一代无线接入网,5G-A迈向6G的智能底座
人工智能·5g·威视锐
Elastic 中国社区官方博客37 分钟前
不再有分配延迟:在无状态 Elasticsearch 中将快照与分片迁移解耦
大数据·运维·人工智能·elasticsearch·搜索引擎·全文检索
俊哥V39 分钟前
每日 AI 研究简报 · 2026-09-02
人工智能·ai
人工智能时代 准备好了吗43 分钟前
搜索别名治理实用指南:在首次出现时说明主名称与别名关系
人工智能
人工智能时代 准备好了吗1 小时前
同名品牌如何避免被AI认错?
人工智能
stolentime1 小时前
OpenClaw 2.0 新手快速上手与实战指南
爬虫·python·ai·网络爬虫
段一凡-华北理工大学1 小时前
高炉智能布料技术与炉料分布优化~专栏简介与目录
android·人工智能·高炉智能化·高炉布料·高炉布料分布·高炉布料参数优化·高炉布料矩阵