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

相关推荐
yiyu07161 分钟前
3分钟搞懂深度学习AI:实操篇:VGG
人工智能·深度学习
金智维科技官方5 分钟前
信通院认证,金智维的政务智能体让政务工作“智能自动化”
人工智能·ai·自动化·agent·智能体
星爷AG I6 分钟前
14-6 运动控制的神经系统(AGI基础理论)
人工智能·agi
猿小猴子10 分钟前
主流 AI IDE 之一的 CodeFlicker 介绍
ide·人工智能
Saniffer_SH10 分钟前
【高清视频】SerialTek PCIe 5.0/6.0 协议分析仪API自动化编程演示
网络·人工智能·驱动开发·嵌入式硬件·测试工具·自动化·压力测试
怪侠_岭南一只猿12 分钟前
爬虫阶段一实战练习题:爬取豆瓣电影 Top250 复盘
css·经验分享·爬虫·python·学习·正则表达式
郭龙_Jack14 分钟前
TensorFlow GPU 优化配置手册
人工智能·python·tensorflow
一个努力编程人17 分钟前
机器学习————GBDT算法
人工智能·算法·机器学习