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

成都創新互聯是一家集網站建設,龍文企業網站建設,龍文品牌網站建設,網站定制,龍文網站建設報價,網絡營銷,網絡優化,龍文網站推廣為一體的創新建站企業，幫助傳統企業提升企業形象加強企業競爭力。可充分滿足這一群體相比中小企業更為豐富、高端、多元的互聯網需求。同時我們時刻保持專業、時尚、前沿，時刻以成就客戶成長自我，堅持不斷學習、思考、沉淀、凈化自己，讓我們為更多的企業打造出實用型網站。

使用tensorflow.python.tools.freeze_graph.freeze_graph

• 使用graph_util.convert_variables_to_constants

1、tensorflow模型的文件解讀

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

checkpoint：記錄近幾次訓練好的模型結果（名稱）。

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

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

xxx.meta：模型的meta數據 ，二進制或者其他格式，不可直接查看，保存了TensorFlow計算圖的結構信息，通俗地講就是神經網絡的網絡結構。

2、最常見的ckpt轉pb文件的方法

2、ckpt轉pb文件（freeze_graph.freeze_graph）

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

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

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

本文名稱：tensorflow模型文件怎么轉pb文件-創新互聯
網頁路徑：http://m.newbst.com/article0/cejgoo.html

成都網站建設公司_創新互聯，為您提供網站營銷、做網站、標簽優化、ChatGPT、搜索引擎優化、微信公眾號

廣告

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