获取tensorflow lite模型指定中间层的输出
以mobilenet v2为例子，我已经训练好了模型，有tflite格式和onnx格式两种模型的权重文件，我想获取tflite模型在推理阶段neck部分的输出。
- 查看
onnx模型得到neck最后一层位置
使用netron查看onnx模型结构
从name中能知道Reshape是neck的最后一层
- 得到
neck最后一层的output在tflite模型中的编号
使用netron查看tflite模型结构找到Reshape的输出的编号
- 在推理脚本中增加保存中间层输出
其他说明：
偶尔同样的操作就会报错ValueError: Tensor data is null. Run allocate_tensors() first，原因有说是tf的版本的原因，但我不更改版本的情况下有时也出现这个问题，以后再细究吧，保存下当前的tf版本：
原文地址：https://blog.csdn.net/weixin_45977690/article/details/136281884
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！