tensorflow学习-如何将pc端训练好的模型转换为移动端可以使用的模型

参考https://www.jianshu.com/p/fa204a54a956
资料

cd D:\software\developUtils\python\Anaconda5.1\Lib\site-packages\tensorflow\python\tools
freeze_graph.py --input_graph=D:\software\github\ai_project\user_portrait\train_model\name2gender.pb --input_checkpoint=D:\software\github\ai_project\user_portrait\train_model\name2gender.model --input_binary=true --out_graph=D:\software\github\ai_project\user_portrait\android_model\name2gender.pb --output_node_names=predictions



D:\software\developUtils\python\Anaconda5.1\Lib\site-packages\tensorflow\tools\graph_transforms\summarize_graph --in_graph=D:\software\github\ai_project\user_portrait\train_model\name2gender.pb

1.找到pc端训练好的模型

tf.train.saver.save一行导出的是模型的变量文件，实际上保存了整个图中所有变量目前的取值。
如图所示，实际上产生了4个文件。在后续步骤中需要用到的是name2gender.model-50.data-00000-of-00001这个文件，保存了当前神经网络各参数的取值

2.生成frozen的graphdef文件

1. 找到pc端tensorflow安装目录

import tensorflow as tf
print(tf.__path__)

3.利用toco（Tensorflow Optimizing COnverter）工具，生成最终的tflite文件。