tensorflow：freeze_graph

说明：freeze_graph，也就是将模型固化，具体讲就是将训练数据和模型固化成pb文件，tensorflow安装目录下有freeze_graph.py文件，该文件可以实现将pb数据文件和ckpt模型文件最终转为一个pb文件，但是实际中网络训练后得到meta、data、index几个文件，如果将这些文件进行固化呢？答案是将该文件简单改下就行，以tensorflow1.2版本为例。

将源文件：

input_graph_def = graph_pb2.GraphDef()

mode = "rb" if input_binary else "r"

with gfile.FastGFile(input_graph, mode) as f:

if input_binary:

input_graph_def.ParseFromString(f.read())

else:

text_format.Merge(f.read(), input_graph_def)

改为：

input_graph_def = meta_graph.read_meta_graph_file(input_graph).graph_def

mode = "rb" if input_binary else "r"

 

具体实现请参考tensorflow源代码中freeze_graph.py