保存和加载pb模型

最新推荐文章于 2024-07-08 08:22:55 发布

MagiChos

最新推荐文章于 2024-07-08 08:22:55 发布

阅读量5.2k

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/hanshuobest/article/details/79222769

版权

将模型保存为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, ["c