如何在tensorflow 2.0中运行冻结图

Question

是否可以在tensorflow 2.0中运行冻结的图？。

在创建会话时观察到以下错误。

with tf.Session(graph=graph) as sess:
   output_data = sess.run(y, feed_dict={x: test_images})

AttributeError：模块'tensorflow'没有属性'Session'

完整的代码可以在这里找到。 Colab notebook

Answer 1

我发现“ tensorflow.python.client”的会话类部分可以在此会话中运行冻结图。

from tensorflow.python.client import session
with session.Session(graph=graph) as sess:
   output_data = sess.run(y, feed_dict={x: test_images})

使用此更新的coLab笔记本