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

相关推荐
Yuezero_2 分钟前
Latent Manifold理论分析
人工智能·算法·机器学习
摸鱼仙人~2 分钟前
自动驾驶经验迁移到AI编码的可行性与方法论研究
人工智能·机器学习·自动驾驶
AI视觉网奇5 分钟前
数字人大模型 daVinci-MagiHuman
人工智能·深度学习
数据与后端架构提升之路5 分钟前
大规模深度学习性能调优:自顶向下的五件套
人工智能·深度学习
子豪-中国机器人8 分钟前
词云支持的 所有核心效果
python
摸鱼仙人~9 分钟前
借鉴自动驾驶运行态安全经验,保障 AI Coding 实时产出安全的方法论研究
人工智能·安全·自动驾驶
ftpeak10 分钟前
LangGraph Agent 开发指南(1~概述)
人工智能·ai·langchain·langgraph
Rkgua11 分钟前
如何让agent禁止访问的某些文件夹呢
人工智能
BlockWay12 分钟前
WEEX与西甲联赛达成2026赛季区域合作
大数据·人工智能
团象科技14 分钟前
跨境出海业务频繁卡壳时,免实名云账号容易踩哪些坑
大数据·人工智能