Caffe predicts same class regardless of image

76 views
Skip to first unread message

Jack Simpson

unread,
Aug 27, 2015, 9:32:22 AM8/27/15
to Caffe Users
modified the MNIST example and when I train it with my 3 image classes it returns an accuracy of 91%. However, when I modify the C++ example with a deploy prototxt file and labels file, and try to test it on some images it returns a prediction of the second class (1 circle) with a probability of 1.0 no matter what image I give it - even if it's images that were used in the training set. I've tried a dozen images and it consistently just predicts the one class. Does anyone know if I've made a mistake in my prototxt files or C++ code for predictions?
Reply all
Reply to author
Forward
0 new messages