diff options
Diffstat (limited to 'tensorflow/examples/label_image/main.cc')
| -rw-r--r-- | tensorflow/examples/label_image/main.cc | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tensorflow/examples/label_image/main.cc b/tensorflow/examples/label_image/main.cc index e830ce3484..8b710e8c1a 100644 --- a/tensorflow/examples/label_image/main.cc +++ b/tensorflow/examples/label_image/main.cc @@ -14,7 +14,8 @@ // customize it to use your own models or images by changing the file names at // the top of the main() function. // -// The googlenet_graph.pb file included by default is created from Inception. +// The model file specified in the README is a pre-trained version of +// an Inception (GoogleNet) model. #include <fstream> @@ -48,12 +49,13 @@ using tensorflow::int32; TF_DEFINE_string(image, "tensorflow/examples/label_image/data/grace_hopper.jpg", "The image to classify (JPEG or PNG)."); -TF_DEFINE_string(graph, - "tensorflow/examples/label_image/data/googlenet_graph.pb", - "The location of the GraphDef file containing the protobuf" - " definition of the network."); +TF_DEFINE_string( + graph, "tensorflow/examples/label_image/data/tensorflow_inception_graph.pb", + "The location of the GraphDef file containing the protobuf" + " definition of the network."); TF_DEFINE_string(labels, - "tensorflow/examples/label_image/data/googlenet_labels.txt", + "tensorflow/examples/label_image/data/" + "imagenet_comp_graph_label_strings.txt", "A text file containing the labels of all the categories, one" " per line."); TF_DEFINE_int32(input_width, 224, "Width of the image the network expects."); |