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

相关推荐
overstarry几秒前
Goland 通过 ACP 连接 Claude Code
人工智能·claude·jetbrains
overstarry几秒前
Midscene.js 初尝试
人工智能·llm
weixin_457760001 分钟前
GIOU (Generalized Intersection over Union) 详解
pytorch·python
汐ya~1 分钟前
Cursor连接异常问题“Model not available/connection failed”解决:IDE内置代理配置,无需全局流量代理与虚拟网卡
python·编辑器·cursor
ZouZou老师2 分钟前
AI时代架构师如何重构研发体系
人工智能
小魔女千千鱼3 分钟前
openEuler AI 开发环境搭建 - Python/Anaconda/Jupyter 完整指南
人工智能
凌晨一点的秃头猪5 分钟前
seed随机种子
人工智能
ChatPPT_YOO6 分钟前
AIPPT工具主题生成深度对比:为什么ChatPPT更胜一筹?
人工智能·信息可视化·powerpoint·ai生成ppt·ppt制作
糖葫芦君6 分钟前
OneRec - V2 lazy decoder为什么效率高
人工智能·深度学习·llm
海市公约7 分钟前
Python操作SQLite数据库:从基础语法到完整项目实战
数据库·ide·python·程序人生·架构·pycharm·sqlite