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

import os
import sys
import numpy as np
import matplotlib.pyplot as plt
caffe_root='/opt/digits-2.0/caffe/'
sys.path.insert(0,caffe_root+'python')
import caffe
MODEL_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()#打印结果