diff --git a/classify_webcam.py b/classify_webcam.py index 40f6b593e..a301e646c 100644 --- a/classify_webcam.py +++ b/classify_webcam.py @@ -31,7 +31,7 @@ def predict(image_data): # Loads label file, strips off carriage return label_lines = [line.rstrip() for line - in tf.gfile.GFile("logs/output_labels.txt")] + in tf.gfile.GFile("logs/trained_labels.txt")] # Unpersists graph from file with tf.gfile.FastGFile("logs/output_graph.pb", 'rb') as f: