91av视频/亚洲h视频/操亚洲美女/外国一级黄色毛片 - 国产三级三级三级三级

  • 大小: 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()??#?返回一個序

評論

共有 條評論