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

相关推荐
hvinsion2 分钟前
Python PDF批量加密工具
android·python·pdf
工业互联网专业9 分钟前
Python大数据可视化:基于Python对B站热门视频的数据分析与研究_flask+hive+spider
hive·python·数据分析·flask·毕业设计·源码·spider
久笙&29 分钟前
高效设计AI Prompt:10大框架详细对比与应用
人工智能·prompt
yuanlulu30 分钟前
mindie推理大语言模型问题及解决方法汇总
人工智能·华为·自然语言处理·nlp·大语言模型·昇腾
学术会议34 分钟前
【智慧光学与高效信号处理】2025年信号处理与光学工程国际会议 (SPOE 2024)
大数据·人工智能·物联网·安全·信号处理
爱研究的小牛37 分钟前
Midjourney技术浅析(一)
人工智能·深度学习·aigc·midjourney
qq_273900231 小时前
PyTorch Lightning Callback介绍
人工智能·pytorch·python
mingo_敏1 小时前
深度学习中的并行策略概述:4 Tensor Parallelism
人工智能·深度学习
游客5201 小时前
设计模式-创建型-工厂方法模式
开发语言·python·设计模式·工厂方法模式
m0_748239831 小时前
Python毕业设计选题:基于django的民族服饰数据分析系统的设计与实现_hadoop+spider
python·django·课程设计