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

相关推荐
paopao_wu2 分钟前
深度学习4:手写数字识别
人工智能·深度学习
CoookeCola3 分钟前
M.I.O: Interactive Intelligence for Digital Humans(交互式智能数字人)
论文阅读·人工智能·aigc·音视频
坐吃山猪5 分钟前
Python命令行工具Click
linux·开发语言·python
那雨倾城5 分钟前
用 YOLO Pose + Segmentation 在PiscCode构建“语义佛光”:一次实时视觉语义融合实验
图像处理·python·opencv·算法·yolo·计算机视觉·视觉检测
人工智能技术咨询.8 分钟前
【无标题】知识图谱构建
人工智能
初九之潜龙勿用9 分钟前
GMM NZ 全流程详解实战:FSDP MOE 训练加速
人工智能·pytorch·python
Clarence Liu9 分钟前
AI Agent开发(1) - eino框架使用
人工智能
山土成旧客9 分钟前
【Python学习打卡-Day28】类的蓝图:从模板到对象的构建艺术
linux·python·学习
爱看科技9 分钟前
亚马逊百亿美元注资OpenAI,微美全息以多模态算力生态抢夺AI模型热潮!
人工智能
架构精进之路13 分钟前
一文搞懂什么是 Vibe Coding?
人工智能·后端