Python tf.compat.v1.saved_model.Builder用法及代码示例

构建 SavedModel 协议缓冲区并保存变量和资产。

用法

tf.compat.v1.saved_model.Builder(
    export_dir
)

SavedModelBuilder 类提供了构建SavedModel 协议缓冲区的函数。具体来说，这允许将多个元图保存为单个 language-neutral SavedModel 的一部分，同时共享变量和资产。

要构建 SavedModel，第一个元图必须与变量一起保存。随后的元图将与其图定义一起保存。如果资产需要保存并写入或复制到磁盘，可以在添加元图def时提供。如果多个元图定义与同名资产相关联，则仅保留第一个版本。

添加到 SavedModel 的每个元图都必须使用标签进行注释。标签提供了一种方法来识别要加载和恢复的特定元图，以及共享的变量和资产集。

SavedModelBuilder 的典型用法：

...
builder = tf.compat.v1.saved_model.Builder(export_dir)

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph_and_variables(sess,
                                  ["foo-tag"],
                                  signature_def_map=foo_signatures,
                                  assets_collection=foo_assets)
...

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  ...
  builder.add_meta_graph(["bar-tag", "baz-tag"])
...

builder.save()

注意：此函数只能通过 v1 兼容性库作为 tf.compat.v1.saved_model.builder.SavedModelBuilder 或 tf.compat.v1.saved_model.Builder 使用。 Tensorflow 2.0 将引入一种基于对象的新方法来创建 SavedModel。