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

相关推荐
YoanAILab3 分钟前
AI 推理系统架构怎么选?图像生成与文本生成的分层选型思路(ComfyUI / Dify / vLLM / Triton)
人工智能·系统架构·comfyui·dify·vllm·ai工程
JavaWeb学起来5 分钟前
Python学习教程(一)环境安装,基本数据类型,变量
开发语言·python·python基础
迷藏4946 分钟前
# 发散创新:用 Rust实现高性能测试框架的底层逻辑与实战演练
java·开发语言·后端·python·rust
林森见鹿8 分钟前
OpenClaw 飞书 ACP 会话配置详解
人工智能
Alter12308 分钟前
华为吴辉:AI正在重构生产系统,“大增量时代”已经到来
人工智能·华为·重构
chushiyunen12 分钟前
python单例模式、大模型一次加载多次复用
开发语言·python·单例模式
pp起床16 分钟前
Part01:大语言模型设置
人工智能·语言模型·自然语言处理
昨夜见军贴061618 分钟前
透析用水与消毒供应中心检测如何更可靠?AI审核与IACheck守住医疗安全底线
人工智能·安全
威联通安全存储19 分钟前
某大型电力装备制造企业:基于威联通NAS的核心图纸保护与数据治理实践
python