TensorFlow模型导出,恢复,转换和api

模型导出
1、h5;
2、ckpt;
3、pb;
4、saved_model;

模型恢复
1、h5,从模型代码构建图,再导入权重;
2、ckpt,从模型代码构建图,再导入权重;或者载入meta_graph,再导入权重;
3、pb,直接载入frozen pb;
4、saved_model,直接载入saved_model;

模型转换
注:pb一般指的是frozen pb,常量化变量用于模型持久化;模型转换的过程中可以添加placeholder,可以用于输入图像的预处理,如resize等;
1、从h5转换成pb;(从模型代码构建图,再导入权重)
2、从h5转换成saved_model;
3、从ckpt转换成pb;(从模型代码构建图,再导入权重;或者载入meta graph,再导入权重)
4、从ckpt转换成saved_model;
5、从pb转换成pb;(直接载入pb,具有完整的网络结构,常量和常量化后的变量)
6、从pb转换成saved_model;
7、从saved_model转换为pb;

三种图(不包括keras的h5)的api
1、SavedModel
保存
tf.saved_model.builder.SavedModelBuilder保存saved_model;
导入
tf.saved_model.loader.load加载saved_model;

2、GraphDef
保存
先使用tf.graph_util.convert_variables_to_constants将Variables常量化,再使用tf.train.write_graph,或者tf.gfile.FastGFile,或者tf.gfile.Gfile保存模型;
导入
tf.import_graph_def

3、CheckPoint
保存
Save
导入
1、先用Import_meta_graph来导入graph,之后再saver.restore();
2、重新建立graph,把tensor传入结构的过程再写一遍,然后再saver.restore();

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值