CKPT 转换成 PB格式

# -*- coding: utf-8 -*-
'''
@author: Carry
@contact: xkx94317@gmail.com
@file: convert.py
@time: 2020/5/19 13:58 
@desc: 
'''
import tensorflow as tf
from tensorflow.python.framework import graph_util
from tensorflow.python import pywrap_tensorflow


def print_tensor_name():
    '''
    获取模型中节点名称
    '''
    checkpoint_path = 'processor/code/DocFace_master/model/faceres_ms/ckpt-320000'
    reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
    var_to_shape_map = reader.get_variable_to_shape_map()
    for key in var_to_shape_map:
        print("tensor_name: ", key)


def freeze_graph(input_checkpoint, output_graph):
    '''
    :param input_checkpoint:
    :param output_graph: PB模型保存路径
    :return:
    '''
    # checkpoint = tf.train.get_checkpoint_state(model_folder) #检查目录下ckpt文件状态是否可用
    # input_checkpoint =