我一直在尝试Tensorflow 2.0 beta
,并且一直在尝试从.pb graphdef
文件中加载模型。
在tensorflow r1.13.1中:
import tensorflow as tf
model_fn = 'graph.pb'
graph = tf.Graph()
sess = tf.InteractiveSession(graph=graph)
with tf.gfile.FastGFile(model_fn, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
但是在tensorflow r2.0 beta中,API完全不同。除了使用compat.v1
模块之外,如何加载.pb文件?
import tensorflow.compat.v1 as tf