模型保存的格式.h5和.pb对比

具有不同特征的不同文件格式,都由tensorflow用于保存模型(.h5特别是keras).

.h5是一种存储结构化数据(在这种情况下是神经网络)的方法,目前已由Google概述.

简单的class包含两个字段,您可以使用多种受支持的语言(例如C++Go)之一将其加载,解析,修改并以二进制格式发送给其他人.

  • 解析力极小且有效(与说出.xml相比),因此通常用于网络上的数据传输
  • 当您要将模型投入生产时(例如推论), Tensorflow的服务使用通过网络)
  • 不可知的语言-二进制格式可以由多种语言(Java,Python,Objective-C和C ++等)读取
  • tf2.0开始建议使用,您可以参阅官方序列化指南
  • 保存各种元数据(使用keras模型的情况下,优化程序,损失等)
  • SavedModel在概念上比单个文件更难掌握
  • 创建weights所在的文件夹
 

keras最初用于保存模型(keras现在正式是tensorflow的一部分)

.pb相比,它不那么通用,更面向数据",编程更少.

  • 用于保存巨型数据(因此某些神经网络很适合)
  • 常用文件保存格式
  • 所有内容都保存在一个文件中(权重,损失,与keras一起使用的优化程序等)
  • 不能与Tensorflow Serving一起使用,但是您可以通过keras.experimental.export_saved_model(model, 'path_to_saved_model')
  • 轻松将其转换为.pb 

如果不需要生产模型(或者距离合理很远),请使用较简单的模型(.h5).如果要进行生产,或者只想在所有tensorflow提供的工具中采用单一格式进行标准化,则使用.pb

  • 1
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

会发paper的学渣

您的鼓励和将是我前进的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值