获取tensorflow lite模型指定中间层的输出

以mobilenet v2为例子,我已经训练好了模型,有tflite格式和onnx格式两种模型的权重文件,我想获取tflite模型在推理阶段neck部分的输出。

  1. 查看onnx模型得到neck最后一层位置

使用netron查看onnx模型结构


从name中能知道Reshape是neck的最后一层

  1. 得到neck最后一层的output在tflite模型中的编号

使用netron查看tflite模型结构找到Reshape的输出的编号

  1. 在推理脚本中增加保存中间层输出

其他说明:

偶尔同样的操作就会报错ValueError: Tensor data is null. Run allocate_tensors() first,原因有说是tf的版本的原因,但我不更改版本的情况下有时也出现这个问题,以后再细究吧,保存下当前的tf版本:

相关推荐
破芽5 分钟前
python知识点整理02
开发语言·python
edtoplort12 分钟前
OpenAI紧急叫停GPT-6.1 Astra训练,奥尔特曼为何踩刹车
人工智能·gpt·算法
打工仔折腾 AI25 分钟前
Docker镜像分层与卷挂载到底怎么工作:一次文件系统层面的实测分析
运维·人工智能·后端·python·docker·容器·性能优化
老歌老听老掉牙27 分钟前
麻花钻锥面后刀面数学模型的完整推导与数值求解
python·钻头·后刀面
墨染天姬34 分钟前
【人工智能训练师】python语法二
开发语言·人工智能·python
深蓝AI1 小时前
748GB 统一内存装进桌面:英伟达 DGX Station 本地跑万亿参数模型,瓶颈不在算力在插座
人工智能·agent
成旭先生1 小时前
AI 文本审核 API:一段中文文本判风险等级、命中标签与处置建议
java·前端·人工智能·api接口·内容风控·文本审核·ugc审核
草上飞95272 小时前
把前沿能力装进便宜产物,本身是多数模型还不会的能力
人工智能·深度学习·llm
林澈在路上2 小时前
2026生成后可发行的AI音乐工具怎么选
人工智能·版权·ai音乐·音乐发行·melo音乐
LoneEon2 小时前
CentOS7 部署 Nacos3.x 集群实战:从注册中心到 AI 管理中心
linux·人工智能·nacos