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

相关推荐
2501_93332955几秒前
媒介宣发技术中台架构实践:基于AI多模态的舆情处置与智能分发系统设计
人工智能·架构·系统架构
带娃的IT创业者9 分钟前
WeClaw WebSocket 连接中断诊断:从频繁掉线到稳定长连的优化之路
python·websocket·网络协议·php·fastapi·实时通信
安全菜鸟10 分钟前
OpenClaw-CN 完整安装教程与避坑指南(国内镜像加速版)
人工智能·openclaw
小慧教你用AI12 分钟前
OpenClaw的多Agent架构设计,揭示其实现原理
人工智能
GinoWi12 分钟前
Chapter 5 Python中的元组
python
FluxMelodySun21 分钟前
机器学习(二十三) 密度聚类与层次聚类
人工智能·机器学习·聚类
奋斗中的小猩猩22 分钟前
Test Case Generator / AI 测试用例生成器(多Agent组合,效果可观)
人工智能·测试用例
总有刁民想爱朕ha22 分钟前
OpenCV 图像操作入门:从零开始玩转计算机视觉
人工智能·opencv·计算机视觉
前进的李工29 分钟前
LangChain使用之Model IO(提示词模版之PromptTemplate)
开发语言·人工智能·python·langchain
Techblog of HaoWANG33 分钟前
目标检测与跟踪(9)-- Jetson Xavier NX GPIO控制3D结构光C与Python双版本实现(中)
c语言·人工智能·目标检测