如何检查张量流保存模型中的图形定义

Posted 2023-02-16

【中文标题】如何检查张量流保存模型中的图形定义

【英文标题】：How to check graph definition in a tensorflow saved model

【发布时间】：2020-12-16 20:51:21

【问题描述】：

我正在使用来自 tensorflow 对象检测模型动物园的一些模型，并使用 tf-trt 进行优化。

我的问题与是否有办法以图形方式或至少以编程方式检查模型的图形定义（逐层）。我的目标是比较原始网络定义和执行后的 trt 优化。

提前感谢您的帮助！

【参考方案1】：

您可以尝试使用 Netron 应用程序来可视化您的图表。 只需使用浏览器会话从netron github repository 上传您的模型。