当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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


从标签指定的 SavedModel 加载模型。 (已弃用)

用法

tf.compat.v1.saved_model.load(
    sess, tags, export_dir, import_scope=None, **saver_kwargs
)

参数

  • sess 用于恢复变量的 TensorFlow 会话。
  • tags 一组字符串标签，用于标识所需的 MetaGraphDef。这些应与使用 SavedModel save() API 保存变量时使用的标签相对应。
  • export_dir SavedModel 协议缓冲区和要加载的变量所在的目录。
  • import_scope 可选的string-- 如果指定，在所有加载的张量名称前面加上 '/'。此范围适用于加载到传递的会话中的张量实例，但它是不是写入静态MetaGraphDef返回的协议缓冲区。
  • **saver_kwargs 传递给 Saver 的可选关键字参数。

返回

  • 在提供的会话中加载的 MetaGraphDef 协议缓冲区。这可用于进一步提取signature-defs、collection-defs等。

抛出

  • RuntimeError 找不到与标签关联的 MetaGraphDef。

迁移到 TF2

警告：这个 API 是为 TensorFlow v1 设计的。继续阅读有关如何从该 API 迁移到本机 TensorFlow v2 等效项的详细信息。见TensorFlow v1 到 TensorFlow v2 迁移指南有关如何迁移其余代码的说明。

tf.compat.v1.saved_model.load 或 tf.compat.v1.saved_model.loader.load 与即刻执行不兼容。请改用tf.saved_model.load 来加载您的模型。您可以参考 SavedModel 指南以获取更多信息以及 tf.saved_model.load 文档字符串中的“从 TensorFlow 1.x 导入 SavedModels”。

如何映射参数

TF1 参数名称 TF2 参数名称 注意
sess 不支持 -
tags tags -
export_dir export_dir -
import_scope 不支持 不需要名称范围。默认情况下，变量与加载的对象相关联，函数名称是重复的。
saver_kwargs 不支持 -

使用示例之前和之后

前：

with tf.compat.v1.Session(graph=tf.Graph()) as sess:
  tf.compat.v1.saved_model.loader.load(sess, ["foo-tag"], export_dir)

后：

model = tf.saved_model.load(export_dir, tags=["foo-tag"])

警告：此函数已弃用。它将在未来的版本中删除。更新说明：此函数只能通过 v1 兼容库作为 tf.compat.v1.saved_model.loader.load 或 tf.compat.v1.saved_model.load 使用。在 Tensorflow 2.0 中将有一个新的函数用于导入 SavedModels。

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.saved_model.load。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。