报错Attempted to use a closed Session.是怎么回事呀

来源:3-9 使用前端界面调用flask发布的ckpt模型(2)

qq_格格BLUE_0

2019-09-19 23:16

我的session应该是正常的呀

x = tf.compat.v1.placeholder('float', [None, 784])
sess = tf.Session()


#取出线性回归方法
with tf.variable_scope('regression'):
    y1, variables = model.regression(x)

saver = tf.train.Saver(variables)
saver.restore(sess, 'mnist/data/regression.ckpt')

#取出训练好的卷积的模型
with tf.variable_scope('convolutionoal'):
    keep_prob = tf.placeholder('float')
    y2, variables = model.convolutional(x, keep_prob)

saver = tf.train.Saver(variables)
#saver.restore(sess, "mnist/data/convalutional.ckpt")
#tf.train.latest_checkpoint('mnist/data/convalutional.ckpt')
module_file =  tf.train.latest_checkpoint('mnist/data/convolutional.ckpt')
with tf.Session() as sess:
   sess.run(tf.global_variables_initializer())
   if module_file is not None:
      saver.restore(sess, module_file)

#定义输入  线性
def regression(input):
    return sess.run(y1, feed_dict={x: input}).flatten().tolist()
#定义输入   卷积
def convolutional(input):
    return sess.run(y2, feed_dict={x: input, keep_prob: 1.0}).flatten().tolist()


写回答 关注

1回答

TensorFlow与Flask结合打造手写体数字识别

TensorFlow和flask结合识别自己的手写体数字

20428 学习 · 102 问题

查看课程

相似问题