1、模型的保存:
import tensorflow as tf v1 = tf.Variable(1.0,dtype=tf.float32) v2 = tf.Variable(2.0,dtype=tf.float32) x = v1 + v2 saver = tf.train.Saver() with tf.Session() as sess: sess.run(tf.global_variables_initializer()) result = sess.run(x) #将模型保存在model文件夹下 saver.save(sess,'./model/test.model') print('result:{}'.format(result))
2、模型的加载(直接加载图)
import tensorflow as tf saver = tf.train.import_meta_graph('./model/test.model.meta') with tf.Session() as sess: saver.restore(sess,'./model/test.model') print(sess.run(tf.get_default_graph().get_tensor_by_name('add:0')))
3、模型的加载(给定映射关系,主要用于不同开发之间模型的调用)
import tensorflow as tf a = tf.Variable(5.0,dtype=tf.float32,name='a') b = tf.Variable(6.0,dtype=tf.float32,name='b') x = a + b saver = tf.train.Saver({'v1':a,'v2':b}) with tf.Session() as sess: saver.restore(sess,'./model/test.model') print(sess.run([x]))