這篇文章主要為大家展示了tensorflow模型文件怎么轉(zhuǎn)pb文件，內(nèi)容簡而易懂，希望大家可以學(xué)習(xí)一下，學(xué)習(xí)完之后肯定會有收獲的，下面讓小編帶大家一起來看看吧。

成都創(chuàng)新互聯(lián)是一家集網(wǎng)站建設(shè),龍文企業(yè)網(wǎng)站建設(shè),龍文品牌網(wǎng)站建設(shè),網(wǎng)站定制,龍文網(wǎng)站建設(shè)報價,網(wǎng)絡(luò)營銷,網(wǎng)絡(luò)優(yōu)化,龍文網(wǎng)站推廣為一體的創(chuàng)新建站企業(yè)，幫助傳統(tǒng)企業(yè)提升企業(yè)形象加強企業(yè)競爭力?？沙浞譂M足這一群體相比中小企業(yè)更為豐富、高端、多元的互聯(lián)網(wǎng)需求。同時我們時刻保持專業(yè)、時尚、前沿，時刻以成就客戶成長自我，堅持不斷學(xué)習(xí)、思考、沉淀、凈化自己，讓我們?yōu)楦嗟钠髽I(yè)打造出實用型網(wǎng)站。

使用tensorflow.python.tools.freeze_graph.freeze_graph

• 使用graph_util.convert_variables_to_constants

1、tensorflow模型的文件解讀

使用tensorflow訓(xùn)練好的模型會自動保存為四個文件，如下

checkpoint：記錄近幾次訓(xùn)練好的模型結(jié)果（名稱）。

xxx.data-00000-of-00001: 模型的所有變量的值(weights, biases, placeholders,gradients, hyper-parameters etc)，也就是模型訓(xùn)練好參數(shù)和其他值。

xxx.index ：模型的元數(shù)據(jù)，二進制或者其他格式，不可直接查看 。是一個不可變得字符串表，每一個鍵都是張量的名稱，它的值是一個序列化的BundleEntryProto。 每個BundleEntryProto描述張量的元數(shù)據(jù)：“數(shù)據(jù)”文件中的哪個文件包含張量的內(nèi)容，該文件的偏移量，校驗和一些輔助數(shù)據(jù)等。

xxx.meta：模型的meta數(shù)據(jù) ，二進制或者其他格式，不可直接查看，保存了TensorFlow計算圖的結(jié)構(gòu)信息，通俗地講就是神經(jīng)網(wǎng)絡(luò)的網(wǎng)絡(luò)結(jié)構(gòu)。

2、最常見的ckpt轉(zhuǎn)pb文件的方法

2、ckpt轉(zhuǎn)pb文件（freeze_graph.freeze_graph）

此種方法嘗試成功，雖然不知道輸出節(jié)點名，但是只要模型代碼還在就可以操作，直接上代碼。

import tensorflow as tf
import os
from tensorflow.python.tools import freeze_graph
from model import network # network是你們自己定義的模型結(jié)構(gòu)（代碼結(jié)構(gòu)）
# egs:
# def network(input)：
# return tf.layers.softmax(input)
 
model_path = "model.ckpt-0000" #設(shè)置model的路徑，因新版tensorflow會生成三個文件，只需寫到數(shù)字前
 
def main():
 tf.reset_default_graph()
 # 設(shè)置輸入網(wǎng)絡(luò)的數(shù)據(jù)維度，根據(jù)訓(xùn)練時的模型輸入數(shù)據(jù)的維度自行修改
 input_node = tf.placeholder(tf.float32, shape=(None, None, 200)) 
 output_node = network(input_node) # 神經(jīng)網(wǎng)絡(luò)的輸出
 # 設(shè)置輸出數(shù)據(jù)類型（特別注意，這里必須要跟輸出網(wǎng)絡(luò)參數(shù)的數(shù)據(jù)格式保持一致，不然會導(dǎo)致模型預(yù)測  精度或者預(yù)測能力的丟失）以及重新定義輸出節(jié)點的名字（這樣在后面保存pb文件以及之后使用pb文件時直接使用重新定義的節(jié)點名字即可）
 flow = tf.cast(output_node , tf.float16, 'the_outputs') 
 saver = tf.train.Saver()
 with tf.Session() as sess:
 saver.restore(sess, model_path)
 #保存模型圖（結(jié)構(gòu)），為一個json文件
 tf.train.write_graph(sess.graph_def, 'output_model/pb_model', 'model.pb')
 #將模型參數(shù)與模型圖結(jié)合，并保存為pb文件
 freeze_graph.freeze_graph('output_model/pb_model/model.pb', '', False, model_path, 'the_outputs','save/restore_all', 'save/Const:0', 'output_model/pb_model/frozen_model.pb', False, "")
 print("done")
if __name__ == '__main__':
 main()

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場景需求。

本文名稱：tensorflow模型文件怎么轉(zhuǎn)pb文件-創(chuàng)新互聯(lián)
網(wǎng)頁路徑：http://chinadenli.net/article0/cejgoo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站營銷、做網(wǎng)站、標(biāo)簽優(yōu)化、ChatGPT、搜索引擎優(yōu)化、微信公眾號

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)