加载.pb模型

Question

我正在尝试加载保存为.pb的预训练模型。 我正在使用以下代码：

从numpy导入loadtxt 从keras.models导入load_model 导入操作系统

def load_pb(path_to_pb):
with tf.gfile.GFile(path_to_pb, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def, name='')
    return graph

然后当我尝试调用它时 tf_graph = load_pb（r'C：\ Users \ karol \ OneDrive \ Desktop \ xxxx \ yyfolder \ saved_model.pb'） sess = tf.Session（graph = tf_graph）

我收到解码器错误：

---

DecodeError跟踪（最近一次通话最近） 在 12返回图 13 ---> 14 tf_graph = tf_graph = load_pb（r'C：\ Users \ karol \ OneDrive \ Desktop \ xxxx \ yyfolder \ saved_model.pb'） 15 sess = tf.Session（graph = tf_graph）

DecodeError：标签中的导线类型错误。

现在有人修复它了吗？