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

相关推荐
田里的水稻几秒前
FA_建图和定位(ML)-超宽带(UWB)定位
人工智能·算法·数学建模·机器人·自动驾驶
罗政1 分钟前
AI批量识别社保卡信息实战:一键提取姓名、卡号、银行账号到Excel
人工智能
遨游xyz2 分钟前
Trie树(字典树)
开发语言·python·mysql
重启编程之路10 分钟前
AlphaLens Pro V14.0 商业级量化推演终端 | 功能白皮书
python
liliangcsdn13 分钟前
基于似然比的显著图可解释性方法的探索
人工智能·算法·机器学习
正宗咸豆花14 分钟前
Gemini 3.1 Pro架构深度解析与AGI能力评测
人工智能·架构·agi
二十雨辰17 分钟前
[python]-生成器和正则
python
systeminof17 分钟前
OpenClaw极速进化:20天迭代10余次,接入Gemini 3.1与Apple Watch
人工智能·chatgpt·ai编程
新缸中之脑17 分钟前
AI调试的3个调查
人工智能
人机与认知实验室21 分钟前
2026:人形机器人的未来发展趋势
大数据·人工智能·机器人