最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

使用keras和tensorflow保存為可部署的pb格式

 更新時(shí)間:2020年05月25日 09:58:14   作者:Mrzhang先森  
這篇文章主要介紹了使用keras和tensorflow保存為可部署的pb格式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

Keras保存為可部署的pb格式

加載已訓(xùn)練好的.h5格式的keras模型

傳入如下定義好的export_savedmodel()方法內(nèi)即可成功保存

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會(huì)自動(dòng)保存為pb格式
 '''
  model_path = "model/" # 模型保存的路徑
  model_version = 0 # 模型保存的版本
  # 從網(wǎng)絡(luò)的輸入輸出創(chuàng)建預(yù)測(cè)的簽名
  model_signature = tf.saved_model.signature_def_utils.predict_signature_def(
    inputs={'input': model.input}, outputs={'output': model.output})
  # 使用utf-8編碼將 字節(jié)或Unicode 轉(zhuǎn)換為字節(jié)
  export_path = os.path.join(compat.as_bytes(model_path), compat.as_bytes(str(model_version))) # 將保存路徑和版本號(hào)join
  builder = tf.saved_model.builder.SavedModelBuilder(export_path) # 生成"savedmodel"協(xié)議緩沖區(qū)并保存變量和模型
  builder.add_meta_graph_and_variables( # 將當(dāng)前元圖添加到savedmodel并保存變量
    sess=K.get_session(), # 返回一個(gè) session 默認(rèn)返回tf的sess,否則返回keras的sess,兩者都沒(méi)有將創(chuàng)建一個(gè)全新的sess返回
    tags=[tf.saved_model.tag_constants.SERVING], # 導(dǎo)出模型tag為SERVING(其他可選TRAINING,EVAL,GPU,TPU)
    clear_devices=True, # 清除設(shè)備信息
    signature_def_map={ # 簽名定義映射
      tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: # 默認(rèn)服務(wù)簽名定義密鑰
        model_signature # 網(wǎng)絡(luò)的輸入輸出策創(chuàng)建預(yù)測(cè)的簽名
    })
  builder.save() # 將"savedmodel"協(xié)議緩沖區(qū)寫入磁盤.
  print("save model pb success ...")

model = keras.models.load_model('model_data/weight.h5') # 加載已訓(xùn)練好的.h5格式的keras模型
export_savedmodel(model) # 將模型傳入保存模型的方法內(nèi),模型保存成功.

Tensorflow保存為可部署的pb格式

1、在tensorflow繪圖的情況下,使用tf.saved_model.simple_save()方法保存模型

2、傳入session

3、傳入保存路徑

4、傳入輸入占位符在inputs={“input_name”: 網(wǎng)絡(luò)輸入占位符變量}

5、傳入輸出變量在outputs={“output_name1”: 網(wǎng)絡(luò)輸出變量, “output_name2”: 網(wǎng)絡(luò)輸出變量}

即可成功保存為可部署的pb格式

tf.saved_model.simple_save(sess,
      "./model",
      inputs={"myInput": x}, # input_name可自定義,編碼客戶端時(shí)對(duì)應(yīng)即可
      outputs={"myOutput": y})

保存好模型后會(huì)得到這樣格式文件證明你保存沒(méi)有問(wèn)題了

variables/
  variables.data-*****-of-*****
  variables.index
saved_model.pb
print_r('點(diǎn)個(gè)贊吧');
var_dump('點(diǎn)個(gè)贊吧');
NSLog(@"點(diǎn)個(gè)贊吧!")
System.out.println("點(diǎn)個(gè)贊吧!");
console.log("點(diǎn)個(gè)贊吧!");
print("點(diǎn)個(gè)贊吧!");
printf("點(diǎn)個(gè)贊吧!\n");
cout << "點(diǎn)個(gè)贊吧!" << endl;
Console.WriteLine("點(diǎn)個(gè)贊吧!");
fmt.Println("點(diǎn)個(gè)贊吧!")
Response.Write("點(diǎn)個(gè)贊吧");
alert('點(diǎn)個(gè)贊吧')

補(bǔ)充知識(shí):將Keras保存的HDF5或TensorFlow保存的PB模型文件轉(zhuǎn)化為Inter Openvino使用的IR(.xml & .bin)文件

本blog依據(jù)英特爾官方手冊(cè)《Model Optimizer Developer Guide》 翻譯編寫,經(jīng)博主測(cè)試可用

intel NCS & OpenVINO

英特爾官方的NCS開發(fā)環(huán)境“OpenVINO”使用了名為Intermediate Representation(IR)的網(wǎng)絡(luò)模型,其中.xml文件保存了網(wǎng)絡(luò)的拓?fù)浣Y(jié)構(gòu),而.bin文件以二進(jìn)制方式保存了模型的權(quán)重w與偏差b。

首先我們需要配置Model Optimizer

如果是安裝適用于所有框架的Model Optimizer:

在安裝完OpenVINO后,我們找到以下位置:

<INSTALL_DIR>/deployment_tools/model_optimizer/install_prerequisites

運(yùn)行以下命令:

對(duì)于Linux系統(tǒng):

install_prerequisites.sh

對(duì)于Windows系統(tǒng):

install_prerequisites.bat

如果只安裝適用于特定框架的Model Optimizer:

在安裝完OpenVINO后,我們找到以下位置:

<INSTALL_DIR>/model_optimizer/install_prerequisites

運(yùn)行以下命令:

對(duì)于Caffe (Linux):

install_prerequisites_caffe.sh

對(duì)于Caffe (Windows):

install_prerequisites_caffe.bat

對(duì)于TensorFlow (Linux):

install_prerequisites_tf.sh

對(duì)于TensorFlow (Windows):

install_prerequisites_tf.bat

對(duì)于MXNet (Linux):

install_prerequisites_mxnet.sh

對(duì)于MXNet (Windows):

install_prerequisites_mxnet.bat

對(duì)于Kaldi (Linux):

install_prerequisites_kaldi.sh

對(duì)于Kaldi (Windows):

install_prerequisites_kaldi.bat

對(duì)于ONNX (Linux):

install_prerequisites_onnx.sh

對(duì)于ONNX (Windows):

install_prerequisites_onnx.bat

如果我們要將TensorFlow保存的PB模型轉(zhuǎn)換為IR……

如果我們要將Keras保存的HDF5模型轉(zhuǎn)換為IR……

博主電腦在英特爾返廠維修中 待更新……

以上這篇使用keras和tensorflow保存為可部署的pb格式就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • Python?shapely庫(kù)的具體使用

    Python?shapely庫(kù)的具體使用

    本文主要介紹了Python?shapely庫(kù)的具體使用,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
    2023-06-06
  • Python的Django框架中的數(shù)據(jù)庫(kù)配置指南

    Python的Django框架中的數(shù)據(jù)庫(kù)配置指南

    這篇文章主要介紹了Python的Django框架中的數(shù)據(jù)庫(kù)配置指南,文中舉了Python內(nèi)置的SQLite的示例,需要的朋友可以參考下
    2015-07-07
  • python的常見命令注入威脅

    python的常見命令注入威脅

    不過(guò)下面可是我們開發(fā)產(chǎn)品初期的一些血淋淋的案例,更多的安全威脅可以看看北北同學(xué)的《python hack》PPT,里面提及了不只命令執(zhí)行的威脅,那些都是我們親身經(jīng)歷的代碼
    2013-02-02
  • python一行代碼就能實(shí)現(xiàn)數(shù)據(jù)分析的pandas-profiling庫(kù)

    python一行代碼就能實(shí)現(xiàn)數(shù)據(jù)分析的pandas-profiling庫(kù)

    這篇文章主要為大家介紹了python一行代碼就能實(shí)現(xiàn)數(shù)據(jù)分析的pandas-profiling庫(kù),有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2024-01-01
  • Python獲取SQLite查詢結(jié)果表列名的方法

    Python獲取SQLite查詢結(jié)果表列名的方法

    這篇文章主要介紹了Python獲取SQLite查詢結(jié)果表列名的方法,涉及Python連接及查詢SQLite數(shù)據(jù)庫(kù)的相關(guān)操作技巧,需要的朋友可以參考下
    2017-06-06
  • Python自動(dòng)登錄126郵箱的方法

    Python自動(dòng)登錄126郵箱的方法

    這篇文章主要介紹了Python自動(dòng)登錄126郵箱的方法,涉及Python針對(duì)郵箱操作的相關(guān)技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下
    2015-07-07
  • Python基于遞歸算法實(shí)現(xiàn)的走迷宮問(wèn)題

    Python基于遞歸算法實(shí)現(xiàn)的走迷宮問(wèn)題

    這篇文章主要介紹了Python基于遞歸算法實(shí)現(xiàn)的走迷宮問(wèn)題,結(jié)合迷宮問(wèn)題簡(jiǎn)單分析了Python遞歸算法的定義與使用技巧,需要的朋友可以參考下
    2017-08-08
  • 簡(jiǎn)單且有用的Python數(shù)據(jù)分析和機(jī)器學(xué)習(xí)代碼

    簡(jiǎn)單且有用的Python數(shù)據(jù)分析和機(jī)器學(xué)習(xí)代碼

    Python編程是一種通用的編程語(yǔ)言,開源、靈活、功能強(qiáng)大且易于使用,python最重要的特性之一是其用于數(shù)據(jù)處理和分析任務(wù)的豐富實(shí)用程序和庫(kù)集,這篇文章主要給大家介紹了一些簡(jiǎn)單且有用的Python數(shù)據(jù)分析和機(jī)器學(xué)習(xí)代碼,需要的朋友可以參考下
    2021-07-07
  • Python Dict 到 Dataclass實(shí)現(xiàn)高效數(shù)據(jù)訪問(wèn)與管理的兩種方式(推薦)

    Python Dict 到 Dataclass實(shí)現(xiàn)高效數(shù)據(jù)訪問(wèn)與管理的兩種方式(推薦)

    本文介紹了Python中的字典和DataClass兩種數(shù)據(jù)結(jié)構(gòu),并探討了如何將字典轉(zhuǎn)換為DataClass,字典適用于鍵值對(duì)存儲(chǔ),感興趣的朋友一起看看吧
    2024-12-12
  • python進(jìn)行數(shù)據(jù)預(yù)處理的4個(gè)重要步驟

    python進(jìn)行數(shù)據(jù)預(yù)處理的4個(gè)重要步驟

    在數(shù)據(jù)科學(xué)項(xiàng)目中,數(shù)據(jù)預(yù)處理是最重要的事情之一,本文詳細(xì)給大家介紹python進(jìn)行數(shù)據(jù)預(yù)處理的4個(gè)重要步驟:拆分訓(xùn)練集和測(cè)試集,處理缺失值,處理分類特征和進(jìn)行標(biāo)準(zhǔn)化處理,需要的朋友可以參考下
    2023-06-06

最新評(píng)論

娱乐| 彝良县| 新泰市| 沙河市| 绵阳市| 改则县| 县级市| 永州市| 保亭| 会理县| 吉林市| 临潭县| 隆昌县| 赣榆县| 冷水江市| 铁岭县| 大渡口区| 灵武市| 安丘市| 汨罗市| 长沙县| 剑阁县| 合川市| 新干县| 汝南县| 贵溪市| 建瓯市| 潮安县| 固阳县| 西宁市| 依兰县| 察哈| 叶城县| 宜章县| 江华| 牟定县| 达孜县| 绥中县| 新郑市| 大同市| 崇阳县|