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

评论

共有 条评论