tensorflow 保存和调用模型

最新推荐文章于 2023-03-02 22:00:46 发布

转载最新推荐文章于 2023-03-02 22:00:46 发布 · 827 阅读

本文详细介绍使用TensorFlow保存和加载模型的过程。首先定义模型结构，包括变量和占位符，然后通过会话运行模型并保存。加载模型时，利用saver对象导入元图和参数，最后验证模型加载正确性。

tensorflow 保存和调用模型

https://www.jianshu.com/p/50243b58b9e1

import tensorflow as tf

x1 = tf.placeholder(dtype=tf.float32, shape=[], name = 'x1')

x2 = tf.placeholder(dtype=tf.float32, shape=[], name = 'x2')

w = tf.Variable(tf.constant(2.), name = 'w')

ytmp = tf.multiply(w, x1, name = 'ytmp')

y = tf.add(ytmp, x2, name = 'y')

sess = tf.Session()

sess.run(tf.global_variables_initializer())

print (sess.run(y, feed_dict={x1: 1., x2: 2.}))

saver = tf.train.Saver()

saver.save(sess, 'model/test')

import tensorflow as tf

sess = tf.Session()

# 导入运算图
saver = tf.train.import_meta_graph('model/test.meta')

#加载相应参数
saver.restore(sess, tf.train.latest_checkpoint('model/'))

graph = tf.get_default_graph()

x1 = graph.get_tensor_by_name('x1:0')

x2 = graph.get_tensor_by_name('x2:0')

y = graph.get_tensor_by_name('y:0')

# restore之后不需要执行variable初始化
#sess.run(tf.global_variables_initializer())
print (sess.run(graph.get_tensor_by_name('w:0')))
print (sess.run(y, feed_dict={x1: 1., x2: 2.}))