如何将keras模型导入透明

时间:2019-06-15 09:48:29

标签: python tensorflow keras tf.keras lucid

简短版本:

我想以透明的方式可视化一个keras模型,并且无法将模型转换为透明的格式。

长版本:

我想清楚地研究一个内置于keras中的简单模型(像这样的https://www.tensorflow.org/tutorials的MNIST的CNN)。

此处将模型导入透明的教程 https://colab.research.google.com/drive/1PPzeZi5sBN2YRlBmKsdvZPbfYtZI-pHl#scrollTo=3YKffRa70uGm

告诉我我需要一个“冻结图”。所以我遵循了本教程

https://towardsdatascience.com/freezing-a-keras-model-c2e26cb84a38

将keras模型转换为tf估计器,对其进行了训练,并应用了教程中提供的freeze_graph()方法。

但是结果图似乎没有清晰的输入节点可以使用-我假设是因为该图基于的估计量期望输入函数而不是张量作为输入?

是否有一种方法可以适应此图,使其接受常规tf张量作为输入?

或者,还有另一种方法可以将keras模型训练并保存为冻结图,而不必绕过tf估计量?

0 个答案:

没有答案