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

相关推荐
EZ_Python3 分钟前
告别WPS会员!用Python自制电子发票批量打印排版工具
python·自动化
Promise微笑3 分钟前
2026年Geo优化的底层逻辑:从语义占位到数字信任的范式重构
大数据·人工智能·搜索引擎·重构·ai搜索
写文章的大米4 分钟前
1 分钟读懂:Python 装饰器
python
qq_397752935 分钟前
智能仓储系统整体架构解析:从仓库规划到自动化落地(工程版)
人工智能·智能仓储·自动化立体仓库·智能仓储整体架构·仓储自动化解决方案·wms 系统·wcs 控制系统
JeffDingAI5 分钟前
【CANN训练营】在CANN8.5上体验Hello World开启Ascend C学习
c语言·开发语言·人工智能·学习
三块可乐两块冰9 分钟前
【第二十六周】机器学习笔记二十五
人工智能·笔记·机器学习
2501_9216494910 分钟前
股指期货 API 入门指南:如何获取实时行情与构建交易系统
python·websocket·金融·区块链·restful
一招定胜负12 分钟前
opencv图片处理常见操作
人工智能·opencv·计算机视觉
byzh_rc13 分钟前
[机器学习-从入门到入土] 特征选择
人工智能·机器学习
Hcoco_me14 分钟前
大模型面试题41:RoPE改进的核心目标与常见方法
开发语言·人工智能·深度学习·自然语言处理·transformer·word2vec