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



我有一个单独使用freeze_graph.py创建的.pb文件。除了独立的.pb文件外,文件夹中没有任何文件。到目前为止,我已经找到了一种方法来可视化使用张量板可视化模型并获取网络中涉及的张量。此外,.pb是使用ResNet架构创建的模型。

1.如何使用此文件对我拥有的自定义数据进行预测?如果我能有一个流程管道,那将有很大的帮助。

2.另外,如何在keras中使用.pb文件?

  1. 好吧,您可以在冻结模型中加载,然后定义一个输入管道。这是一篇更详细地描述这一点的文章。我还建议看看Tensorflow Serving。

  2. 不要认为这是有意义的,因为Keras使用HDF5来存储他们的模型。此外,使用单个.pb进行开发也没有任何意义,因为这不是它的用例。如果你想有灵活性,你应该使用检查点,它有图表和某些时期的权重。

最新更新