张量流中的独立协议缓冲区文件及其用法?

时间:2017-08-15 09:08:37

标签: python tensorflow

我有一个.pb文件单独使用freeze_graph.py创建,而不是独立的.pb文件我没有文件夹中的任何文件。直到现在我已经想出了一种使用张量板可视化模型并获取网络中涉及的张量的方法。此外,.pb是使用ResNet架构创建的模型。

1.如何利用此文件对我拥有的自定义数据进行预测?如果我可以拥有一个流程管道,那将是非常有用的。

2.我如何在keras中使用.pb文件?

1 个答案:

答案 0 :(得分:0)

  1. 您可以加载冻结模型然后定义输入管道。这里有an article更详细地描述了这一点。我还建议你看一下Tensorflow Serving

  2. 不要认为这是有道理的,因为Keras使用HDF5来存储他们的模型。使用单个.pb进行开发也没有任何意义,因为这不是它的用例。如果您希望具有灵活性,则应使用具有图表和特定时期权重的检查点。