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

相关推荐
道19934 分钟前
PyTorch 从小白到高级全阶段学习大纲(一)
人工智能·pytorch·学习
我要学脑机9 分钟前
关于脑动态网络研究的一些方法和论文
人工智能
光影少年12 分钟前
前端ai开发需要学习哪些东西?
前端·人工智能·学习
灵途科技29 分钟前
灵途科技当选中国电子商会智能传感器专委会副理事长单位
大数据·人工智能·科技
非著名架构师29 分钟前
“低空经济”的隐形护航者:AI驱动的秒级风场探测如何保障无人机物流与城市空管安全?
人工智能·数据分析·疾风气象大模型·高精度天气预报数据·galeweather.cn·高精度气象
E_ICEBLUE34 分钟前
PDF vs PDF/A:区别、场景与常用转换方法(2025 全面解读)
python·pdf
洁洁!44 分钟前
openEuler在WSL2中的GPU加速AI训练实战指南
人工智能·数据挖掘·数据分析
桂花饼1 小时前
字节Seedream-4.5架构揭秘:当AI开始拥有“版式推理”能力,CISAN与DLE引擎如何重构多图生成?
人工智能·aigc·idea·sora2 api·gemini 3 pro·claude opus 4.5·doubao-seedream
whaosoft-1431 小时前
51c视觉~合集55
人工智能
AI营销快线1 小时前
2025年AI营销内容生产革命:成本减半,效率倍增的关键
人工智能