diff --git a/run_scene.py b/run_scene.py index 2036a3b..529a226 100644 --- a/run_scene.py +++ b/run_scene.py @@ -50,5 +50,9 @@ def classify_scene(fpath_design, fpath_weights, fpath_labels, fpath_mean, image) test_image = caffe.io.load_image(sys.argv[1]) predictions = classify_scene(MODEL_TYPE, WEIGHTS, LABELS, MEAN, test_image) + + with open(LABELS, 'rb') as f: + labels = pickle.load(f) + for i, k in enumerate(predictions): - print i, labels[k] \ No newline at end of file + print i, labels[k]