使用训练好的caffe模型识别图片

import osimport sysimport numpy as npimport matplotlib.pyplot as pltcaffe_root='/opt/digits-2.0/caffe/'sys.path.insert(0,caffe_root+'python')import caffeMODEL_FILE='/opt/digits-2.0/caffe/examples/mnist/lenet.prototxt'#网络模型PRETRAINED='/opt/digits-2.0/caffe/examples/mnist/lenet_iter_10000.caffemodel'#训练好的模型，也就是参数IMAGE_FILE='/home/amax/fp/caffe/examples/mnist/8.jpg'#自己的图片input_image=caffe.io.load_image(IMAGE_FILE,color=False)#caffe接口载入图片net = caffe.Classifier(MODEL_FILE,PRETRAINED)#载入分类器predition=net.predict([input_image],oversample = False)#预测图片进行分类caffe.set_mode_cpu()print 'predictied class:',predition[0].argmax()#打印结果