-
大小: 2KB文件類型: .py金幣: 1下載: 0 次發布日期: 2021-05-06
- 語言: Python
- 標簽: tensorflow??pb??
資源簡介
使用這個python程序可以直接將ckpt文件轉成固化的模型文件,用于預測結果,注意修改路徑
代碼片段和文件信息
import?tensorflow?as?tf
#from?create_tf_record?import?*
from?tensorflow.python.framework?import?graph_util
def?freeze_graph(input_checkpointoutput_graph):
????‘‘‘
????:param?input_checkpoint:
????:param?output_graph:?PB模型保存路徑
????:return:
????‘‘‘
????#?checkpoint?=?tf.train.get_checkpoint_state(model_folder)?#檢查目錄下ckpt文件狀態是否可用
????#?input_checkpoint?=?checkpoint.model_checkpoint_path?#得ckpt文件路徑
?
????#?指定輸出的節點名稱該節點名稱必須是原模型中存在的節點
????output_node_names?=?“InceptionV1/Logits/Predictions/Reshape_1“
????saver?=?tf.train.import_meta_graph(input_checkpoint?+?‘.meta‘?clear_devices=True)
????graph?=?tf.get_default_graph()?#?獲得默認的圖
????input_graph_def?=?graph.as_graph_def()??#?返回一個序
評論
共有 條評論