保存和加载pb模型

最新推荐文章于 2024-06-09 11:30:05 发布

原创

最新推荐文章于 2024-06-09 11:30:05 发布 · 5.3k 阅读

3 ·

CC 4.0 BY-SA版权

将模型保存为pb

import tensorflow as tf
from tensorflow.python.framework import graph_util

logdir='output/'

with tf.variable_scope('conv'):
     w=tf.get_variable('w',[2,2],tf.float32,initializer=tf.random_normal_initializer)
     b=tf.get_variable('b',[2],tf.float32,initializer=tf.random_normal_initializer)


sess=tf.InteractiveSession()

tf.global_variables_initializer().run() # 初始化所有变量
constant_graph_w = graph_util.convert_variables_to_constants(sess, sess.graph_def, ["conv/w"])
constant_graph_b = graph_util.convert_variables_