为了分享我们经过训练的 tensorflow 网络,我们将图形冻结到一个.pb文件中。我们还创建了一个 xml 文件,其中包含一些元数据,例如输入张量和输出张量、要应用的预处理类型、训练数据信息等。然后通过加载图形和评估张量等使用 Java 或 C# 提供模型。
为了使共享更容易,我想在.pb文件中的某处包含此 xml 数据。有没有办法做到这一点?一个想法是将它作为 tf.Constant,但我不知道如何将它连接到普通图。
请注意,这是使用freeze_graph.py. 新的 SavedModel 格式是否更合适?