Testing output labels of sample images

[bilgi...@gmail.com]

Hi everyone,

I train the network and i would like to test output labels of testing images. In training, caffe show the accuracy not output labels. I search on the net and try to get predicted labels with sample python code:

 http://nbviewer.ipython.org/github/BVLC/caffe/blob/master/examples/00-classification.ipynb

but because of this line  

output_prob = output['prob'][0]  # the output probability vector for the first image in the batch

print 'predicted class is:', output_prob.argmax()

I gave Key Error 'prob', output_prob is not include prob attribute