如何将keras模型导入lucid

2024-04-30 00:40:23 发布

您现在位置:Python中文网/ 问答频道 /正文

简短版本:

我希望在lucid中可视化一个keras模型,但未能将模型转换为lucid可以接受的格式。在

长版:

我想和lucid一起探索一个用keras构建的简单模型(CNN for MNIST like thishttps://www.tensorflow.org/tutorials)。在

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

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

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

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

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

有没有一种方法可以调整这个图,让它接受规则的tf张量作为输入?在

或者,是否有另一种方法来训练和保存keras模型为冻结图,而不必绕道tf估计量?在


Tags: 方法https模型版本comfor可视化tf