在另一个函数中使用Tensorflow会话时找不到变量

Question

我正在创建一个Flask应用程序，我只想用它的变量还原tensorflow会话一次，并在各种路径上使用它来进行预测，但它似乎不起作用。我创建了一个类，其构造函数还原了交互式会话及其变量，该类还包含一个名为predict的函数，该函数尝试使用构造函数中描述的交互式会话来预测结果

class MyServer(Flask):
def __init__(self, *args, **kwargs):
        super(MyServer, self).__init__(*args, **kwargs)
        self.testClass=TestClass()
app = MyServer(__name__)

这是我使用的路线的一个例子

@app.route('/')
def index():
    size_dict=app.testClass.test(saveName)
    return jsonify(size_dict)

但这会引发以下错误

ValueError: Variable Net_WT/block1/bn does not exist, or was not created with tf.get_variable(). Did you mean to set reuse=tf.AUTO_REUSE in VarScope?

但是如果我初始化对象并以相同的方式使用它，它就会工作

Answer 1

我不想回答我自己的问题，但我找到了一个解决方案，在还原会话和变量之后，我在TestClass的构造函数中添加了self.graph = tf.get_default_graph()