加载.pb文件

loadModel.py

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
    
#你可以检索感兴趣的张量
input = graph.get_tensor_by_name('input:0')
output = graph.get_tensor_by_name('output:0')

#或者
sess.run(output, feed_dict={input: some_data})