2020
09-29
09-29
使用keras和tensorflow保存为可部署的pb格式
Keras保存为可部署的pb格式加载已训练好的.h5格式的keras模型传入如下定义好的export_savedmodel()方法内即可成功保存importkerasimportosimporttensorflowastffromtensorflow.python.utilimportcompatfromkerasimportbackendasKdefexport_savedmodel(model):'''传入kerasmodel会自动保存为pb格式'''model_path="model/"#模型保存的路径model_version=0#模型保存的版本#从网络的输入输出创建预测...
继续阅读 >