分类：pb模型

2020
10-08

tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例

升级到tf2.0后,训练的模型想转成1.x版本的.pb模型,但之前提供的通过ckpt转pb模型的方法都不可用(因为保存的ckpt不再有.meta)文件,尝试了好久,终于找到了一个方法可以迂回转到1.x版本的pb模型.Note:本方法首先有些要求需要满足:可以拿的到模型的网络结构定义源码网络结构里面的所有操作都是通过tf.keras完成的,不能出现类似tf.nn的tensorflow自己的操作符tf2.0下保存的模型是.h5格式的,并且仅保存了weights,即通过model.sa... 继续阅读 >

2020
09-29

将keras的h5模型转换为tensorflow的pb模型操作

背景：目前keras框架使用简单，很容易上手，深得广大算法工程师的喜爱，但是当部署到客户端时，可能会出现各种各样的bug，甚至不支持使用keras，本文来解决的是将keras的h5模型转换为客户端常用的tensorflow的pb模型并使用tensorflow加载pb模型。h5_to_pb.pyfromkeras.modelsimportload_modelimporttensorflowastfimportosimportos.pathasospfromkerasimportbackendasK#路径参数input_path='inputpath'weight_... 继续阅读 >