多语言展示
当前在线:1484今日阅读:26今日分享:39

python怎么使用训练好的模型

保存tensorflow训练好的模型并调用此模型
工具/原料
1

python

2

tensorflow2.0

保存为HDF5文件
1

保存模型Keras 可以使用 HDF5 标准提供基本保存格式。出于我们的目的,可以将保存的模型视为单个二进制blob,如下图

2

调用模型从保存的文件中重新加载模型调用,如下图

saved_model 保存
1

创建并训练模型

2

保存模型创建一个 saved_model,并将其放在带有 tf.keras.experimental.export_saved_model 的带时间戳的目录中

3

调用模型从保存的模型重新加载新的 Keras 模型

注意事项

save_model这种保存 tf.keras 模型的方法是实验性的,在将来的版本中可能有所改变,不 建议使用

推荐信息