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

相关推荐
ARM+FPGA+AI工业主板定制专家2 分钟前
基于JETSON/RK3588+FPGA+AI农业机器人视觉感知方案
人工智能·计算机视觉·fpga开发·机器人
程序员小远6 分钟前
Web自动化测试详解
自动化测试·软件测试·python·selenium·测试工具·职场和发展·测试用例
lomocode7 分钟前
大模型本地部署与预热全攻略:让首次响应速度提升 5 倍
人工智能
生信大表哥12 分钟前
如何在服务器上使用 Gemini 3 进行生信分析:从入门到进阶
linux·人工智能·语言模型·数信院生信服务器·生信云服务器
某林21212 分钟前
SLAM 建图系统配置与启动架构
人工智能·stm32·单片机·嵌入式硬件·算法
红石程序员14 分钟前
Python环境管理
开发语言·python
cooldream200915 分钟前
深入理解 Cursor 规则体系
人工智能·cursor
Chennnng16 分钟前
关于python版本,显卡版本,torch版本之间的问题
开发语言·python
AI浩22 分钟前
【Qwen3-VL-4B-Instruct实战】推理图片和视频、加速
人工智能
腾飞开源27 分钟前
26_Spring AI 干货笔记之 OCI GenAI Cohere 聊天
人工智能·自动配置·依赖管理·springai·聊天模型·运行时选项·oci cohere