What is the use of a *.pb file in TensorFlow and how does it work?

Question

I am using some implementation for creating a face recognition which uses this file:

"facenet.load_model("20170512-110547/20170512-110547.pb")"

What is the use of this file? I am not sure how it works.

Console log:

Model filename: 20170512-110547/20170512-110547.pb
distance = 0.72212267

GitHub link of the actual owner of the code: https://github.com/arunmandal53/facematch

Answer 1

pb stands for protobuf. In TensorFlow, the protbuf file contains the graph definition as well as the weights of the model. Thus, a pb file is all you need to be able to run a given trained model.

Given a pb file, you can load it as follows:

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

Once you have loaded the graph, you can basically do anything. For instance, you can retrieve tensors of interest with

input = graph.get_tensor_by_name('input:0')
output = graph.get_tensor_by_name('output:0')

and use regular TensorFlow routine like:

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