相关疑难解决方法(0)

将元数据添加到 tensorflow 冻结图 pb

为了分享我们经过训练的 tensorflow 网络,我们将图形冻结到一个.pb文件中。我们还创建了一个 xml 文件,其中包含一些元数据,例如输入张量和输出张量、要应用的预处理类型、训练数据信息等。然后通过加载图形和评估张量等使用 Java 或 C# 提供模型。

为了使共享更容易,我想在.pb文件中的某处包含此 xml 数据。有没有办法做到这一点?一个想法是将它作为 tf.Constant,但我不知道如何将它连接到普通图。

请注意,这是使用freeze_graph.py. 新的 SavedModel 格式是否更合适?

python tensorflow

5
推荐指数
1
解决办法
767
查看次数

标签 统计

python ×1

tensorflow ×1