Keras保存为可部署的pb格式
加载已训练好的.h5格式的keras模型
传入如下定义好的export_savedmodel()方法内即可成功保存
import keras import os import tensorflow as tf from tensorflow.python.util import compat from keras import backend as K def export_savedmodel(model): ''' 传入keras model会自动保存为pb格式 ''' model_path = "model/" # 模型保存的路径 model_version = 0 # 模型保存的版本 # 从网络的输入输出创建预测的签名 model_signature = tf.saved_model.signature_def_utils.predict_signature_def( inputs={'input': model.input}, outputs={'output': model.output}) # 使用utf-8编码将 字节或Unicode 转换为字节 export_path = os.path.join(compat.as_bytes(model_path), compat.as_bytes(str(model_version))) # 将保存路径和版本号join builder = tf.saved_model.builder.SavedModelBuilder(export_path) # 生成"savedmodel"协议缓冲区并保存变量和模型 builder.add_meta_graph_and_variables( # 将当前元图添加到savedmodel并保存变量 sess=K.get_session(), # 返回一个 session 默认返回tf的sess,否则返回keras的sess,两者都没有将创建一个全新的sess返回 tags=[tf.saved_model.tag_constants.SERVING], # 导出模型tag为SERVING(其他可选TRAINING,EVAL,GPU,TPU) clear_devices=True, # 清除设备信息 signature_def_map={ # 签名定义映射 tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: # 默认服务签名定义密钥 model_signature # 网络的输入输出策创建预测的签名 }) builder.save() # 将"savedmodel"协议缓冲区写入磁盘. print("save model pb success ...") model = keras.models.load_model('model_data/weight.h5') # 加载已训练好的.h5格式的keras模型 export_savedmodel(model) # 将模型传入保存模型的方法内,模型保存成功.
Tensorflow保存为可部署的pb格式
1、在tensorflow绘图的情况下,使用tf.saved_model.simple_save()方法保存模型
2、传入session
3、传入保存路径
4、传入输入占位符在inputs={“input_name”: 网络输入占位符变量}
5、传入输出变量在outputs={“output_name1”: 网络输出变量, “output_name2”: 网络输出变量}
即可成功保存为可部署的pb格式
tf.saved_model.simple_save(sess, "./model", inputs={"myInput": x}, # input_name可自定义,编码客户端时对应即可 outputs={"myOutput": y})
保存好模型后会得到这样格式文件证明你保存没有问题了
variables/ variables.data-*****-of-***** variables.index saved_model.pb
print_r('点个赞吧'); var_dump('点个赞吧'); NSLog(@"点个赞吧!") System.out.println("点个赞吧!"); console.log("点个赞吧!"); print("点个赞吧!"); printf("点个赞吧!\n"); cout << "点个赞吧!" << endl; Console.WriteLine("点个赞吧!"); fmt.Println("点个赞吧!") Response.Write("点个赞吧"); alert('点个赞吧')
补充知识:将Keras保存的HDF5或TensorFlow保存的PB模型文件转化为Inter Openvino使用的IR(.xml & .bin)文件
本blog依据英特尔官方手册《Model Optimizer Developer Guide》 翻译编写,经博主测试可用
intel NCS & OpenVINO
英特尔官方的NCS开发环境“OpenVINO”使用了名为Intermediate Representation(IR)的网络模型,其中.xml文件保存了网络的拓扑结构,而.bin文件以二进制方式保存了模型的权重w与偏差b。
首先我们需要配置Model Optimizer
如果是安装适用于所有框架的Model Optimizer:
在安装完OpenVINO后,我们找到以下位置:
<INSTALL_DIR>/deployment_tools/model_optimizer/install_prerequisites
运行以下命令:
对于Linux系统:
install_prerequisites.sh
对于Windows系统:
install_prerequisites.bat
如果只安装适用于特定框架的Model Optimizer:
在安装完OpenVINO后,我们找到以下位置:
<INSTALL_DIR>/model_optimizer/install_prerequisites
运行以下命令:
对于Caffe (Linux):
install_prerequisites_caffe.sh
对于Caffe (Windows):
install_prerequisites_caffe.bat
对于TensorFlow (Linux):
install_prerequisites_tf.sh
对于TensorFlow (Windows):
install_prerequisites_tf.bat
对于MXNet (Linux):
install_prerequisites_mxnet.sh
对于MXNet (Windows):
install_prerequisites_mxnet.bat
对于Kaldi (Linux):
install_prerequisites_kaldi.sh
对于Kaldi (Windows):
install_prerequisites_kaldi.bat
对于ONNX (Linux):
install_prerequisites_onnx.sh
对于ONNX (Windows):
install_prerequisites_onnx.bat
如果我们要将TensorFlow保存的PB模型转换为IR……
如果我们要将Keras保存的HDF5模型转换为IR……
博主电脑在英特尔返厂维修中 待更新……
以上这篇使用keras和tensorflow保存为可部署的pb格式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。
更新动态
- 凤飞飞《我们的主题曲》飞跃制作[正版原抓WAV+CUE]
- 刘嘉亮《亮情歌2》[WAV+CUE][1G]
- 红馆40·谭咏麟《歌者恋歌浓情30年演唱会》3CD[低速原抓WAV+CUE][1.8G]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[320K/MP3][193.25MB]
- 【轻音乐】曼托凡尼乐团《精选辑》2CD.1998[FLAC+CUE整轨]
- 邝美云《心中有爱》1989年香港DMIJP版1MTO东芝首版[WAV+CUE]
- 群星《情叹-发烧女声DSD》天籁女声发烧碟[WAV+CUE]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[FLAC/分轨][748.03MB]
- 理想混蛋《Origin Sessions》[320K/MP3][37.47MB]
- 公馆青少年《我其实一点都不酷》[320K/MP3][78.78MB]
- 群星《情叹-发烧男声DSD》最值得珍藏的完美男声[WAV+CUE]
- 群星《国韵飘香·贵妃醉酒HQCD黑胶王》2CD[WAV]
- 卫兰《DAUGHTER》【低速原抓WAV+CUE】
- 公馆青少年《我其实一点都不酷》[FLAC/分轨][398.22MB]
- ZWEI《迟暮的花 (Explicit)》[320K/MP3][57.16MB]