主要介绍了tensorflow使用freeze_graph.py将ckpt转为pb文件的方法,需要的朋友可以参考下

废话少说直接上代码样例如下 import tensorflow as tf import os from tensorflow.python.tools import freeze_graph # 本来这个model本无需解释太多，但是这么多人不能耐下心来看，那么我简单的说一下吧 # network是你们自己定义的模型结构而已 # ps: # def network(input)： # return tf.layers.max_pooling2d(input, 2, 2) from model