如何在Tensorflow 2.0 Beta中加载冻结的图.pb文件?

时间:2019-06-17 11:41:43

标签: python tensorflow

我一直在尝试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

0 个答案:

没有答案