讲师中心 微信公众号
AI工具推荐 视频效率加速

在Python中如何保存与加载TensorFlow的SavedModel

秋瑶君_7897

秋瑶君_7897

发布时间:2026-11-07 10:45:36

|

510人浏览过

|

来源于php中文网

原创

tf.keras.models.save_model 保存模型必须显式指定 save_format="tf" 并传入目录路径,含自定义对象需同步提供 custom_objects;加载必须用 tf.keras.models.load_model 且再次传入 custom_objects,否则报错。

在python中如何保存与加载tensorflow的savedmodel

用 tf.keras.models.save_model 保存模型,不是 model.save() 的所有重载都等价

直接调用 model.save("path/to/model") 默认行为取决于模型类型和 TensorFlow 版本:Keras 模型在 TF 2.4+ 中默认保存为 SavedModel 格式,但若传入 save_format="h5" 或后缀为 .h5,就会走 HDF5 路径——这会导致后续无法用 tf.keras.models.load_model 正确加载为 SavedModel(尤其含自定义层、符号张量或 tf.function 的模型)。稳妥做法是显式指定格式:

  • tf.keras.models.save_model(model, "my_model", save_format="tf") —— 强制输出 SavedModel 目录
  • 路径必须是**目录名**(不能带 .h5 或 .pb 后缀),例如 "./saved_model_dir",内部会生成 assets/、variables/ 和 saved_model.pb
  • 若模型含自定义对象(如自定义层、损失函数),需同时传入 custom_objects=... 参数,否则保存虽成功,加载时会报 ValueError: Unknown layer

加载 SavedModel 必须用 tf.keras.models.load_model,不能用 tf.saved_model.load 直接当 Keras 模型用

tf.saved_model.load("path/to/model") 返回的是一个 ConcreteFunction 或模块化签名对象,它没有 .predict()、.fit() 等 Keras 接口,也无法直接参与训练流程。只有 tf.keras.models.load_model 才能重建完整的 Keras 模型实例(含权重、结构、编译配置):

  • 加载时若原模型用了自定义对象,必须再次传入 custom_objects,且键名要和定义时一致,比如 {"MyCustomLayer": MyCustomLayer}
  • 如果只关心推理,且模型有明确签名(如 "serving_default"),可用 loaded = tf.saved_model.load("path"); infer = loaded.signatures["serving_default"],但这返回的是纯函数,输入需是 tf.Tensor,不支持 numpy 数组或 batched list
  • 加载后的 Keras 模型默认处于 inference 模式;若需继续训练,确认它已编译(loaded.compile(...)),否则调 .fit() 会报 RuntimeError: The model needs to be compiled before calling fit

保存/加载时 shape 不匹配或 dtype 报错,大概率是输入签名没对齐

SavedModel 在保存时会固化输入张量的 shape 和 dtype(尤其是通过 model.predict() 首次调用后自动推导,或显式用 tf.TensorSpec 定义签名)。加载后若喂入不同 shape 的数据(如训练时 batch size=32,推理时送入 batch=1 但未设 None 维度),会触发 InvalidArgumentError: Input to reshape is a tensor with X values, but the requested shape has Y。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • 保存前确保模型输入层 shape 含动态维度,例如 Input(shape=(None, 128)) 或 Input(shape=(None, None, 3)),避免写死 batch 维度
  • 更可靠的方式是在保存时显式指定签名:tf.keras.models.save_model(model, "my_model", signatures=model.call.get_concrete_function(tf.TensorSpec(shape=[None, 224, 224, 3], dtype=tf.float32)))
  • 加载后检查输入签名:loaded.signatures["serving_default"].structured_input_signature,对比你实际输入的数据结构

跨 Python 或 TF 版本加载失败,重点查 saved_model.pb 兼容性边界

SavedModel 格式本身向后兼容(高版本 TF 可加载低版本保存的模型),但反向不保证。TF 2.8 保存的模型在 TF 2.5 上可能因 opset 升级而报 Op type not registered;Python 版本差异通常不影响,但若模型里硬编码了 3.9 特有的语法(如带括号的字面量表达式),则加载时会因 custom_objects 导入失败间接报错。

立即学习“Python免费学习笔记(深入)”;

  • 生产部署建议固定 TF 版本,并在 CI 中用目标环境验证加载逻辑
  • 不要依赖 tf.__version__ 字符串做条件分支,SavedModel 内部不存 Python 版本信息,只存计算图与变量值
  • 若必须降级加载,唯一可行路径是:用原 TF 版本加载 → 重新 save_model → 在低版本中加载新保存的模型
SavedModel 的“透明性”是假象——它看似封装完整,实则把输入约束、自定义依赖、版本隐含假设全打包进二进制文件里。最容易被忽略的是签名固化时机和 custom_objects 的两次一致性:保存时没传,可能侥幸成功;加载时漏传,必报错,而且错误信息往往不指明缺哪个 key。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

Flowith
Flowith Hot

Flowith是一款基于画布和智能体工作流的 AI 协作工作空间。

讯飞智文

一款面向学习和办公场景的AI文档创作工具,可辅助生成PPT与Word文档,提高资料整理和内容制作效率。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1891

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

5164

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1829

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

31177

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3687

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3707

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1243

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

616

2023.08.10

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

20

2026.11.05

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn