一.将tensorflow的ckpt模型转换成pbtxt模型
转换模型格式的代码可参考如下demo:
#! /usr/bin/python
# _*_ coding: utf-8 _*_
__author__ = 'Jeffery'
__date__ = '2018/12/29 23:20'
import tensorflow as tf
from mymodel import captcha_model as model
import os
from constant import char_num, classes
def export_model(checkpoint_path,
export_model_dir,
model_version
):
"""
:param checkpoint_path: type string, original model path(a dir)
:param export_model_dir: type string, save dir for exported model
:param model_version: type int best
:return:no return
"""
with tf.get_default_graph().as_default():
input_images = tf.placeholder(tf.float32, shape=[None, 100, 120, 1], name='input_images')
output_result, _ = model(input_images, keep_prob=1.0, trainable=False)
output_result = tf.argmax(tf.reshape(output_result, [-1, char_num, classes]), 2)
saver = tf.train.Saver()
with tf.Session() as sess:
ckpt_state = tf.train.get_checkpoint_state(checkpoint_path)
model_path = os.path.join(checkpoint_path,
os.path.basename