汉字分类器

ancientcc

花了几天复习tensorflow，终于可以用你的模型了。从百度网盘下载模型，由它们生成C++可运行的*.pb时却报错了。

以下是生成*.pb的代码。关于更多把训练出的模型保存成C++可运行的*.pb格式，参考C++运行TensorFlow模型的“五、保存训练出的模型”。

1. import tensorflow as tf;
   
2. from tensorflow.python.framework import graph_util;
   
3. 
   
4. saver = tf.train.import_meta_graph("/mnt/MNIST_model/chinese-rec-model-16001.meta");
   
5. 
   
6. with tf.Session() as sess:
   
7.         saver.restore(sess, "/mnt/MNIST_model/chinese-rec-model-16001");
   
8. 
   
9.         graph_def = tf.get_default_graph().as_graph_def();
   
10.         output = graph_util.convert_variables_to_constants(sess, graph_def, ['fc2/logits']);
    
11.         with tf.gfile.GFile("/mnt/MNIST_model/combined_model.pb", "wb") as f:
    
12.                  f.write(output.SerializeToString());
    
13. 
    

复制代码

安装的是Windows 10，是通过docker安装的tensroflow 0.12.0。从github下载chinese_character_recognition_bn.py，能够训练。不过14轮后被系统Killed，我估计是内存等原因，毕竟我的PC就是个8G的笔记本。运行用的命令是你在github的示例命令。

1. python chinese_character_recognition_bn.py --mode=train --max_steps=16002 --eval_steps=100 --save_steps=500

复制代码

这是训练时控制台输出情况。