正确将模型保存到.pb

Question

我使用了教程“ Classify Flowers with Transfer Learning”来重新训练自己的课程。 不幸的是，没有提供有关如何将模型正确保存为.pb的示例。

我试图根据我可以阅读的不同文章来实现我的解决方案。最后，我将某些内容另存为.pb，但以后无法重新打开模型。我收到以下错误：

var arrayResults = collection.find({'productType':'kcc'});

下面，在**之间是我为保存模型而修改的上述colab笔记本的各个部分：

ValueError: Input 0 of node AssignVariableOp was passed float from
module/InceptionV3/Conv2d_1a_3x3/BatchNorm/beta:0 inco
mpatible with expected resource

Answer 1

这是可行的，但可能有点儿怪异，这就是为什么tensorflow附带了为您执行此操作的工具。

请参阅： freeze_graph.py

您可以深入研究它的工作原理，也可以直接导入并使用它。