利用 caffe的 python接口测试训练好的 mnist 模型

最新推荐文章于 2019-05-25 23:53:48 发布

pomodoro_m

最新推荐文章于 2019-05-25 23:53:48 发布

阅读量297

点赞数 1

分类专栏： caffe 学习文章标签： caffe python mnist windows

本文链接：https://blog.csdn.net/weixin_42970026/article/details/84399324

版权

caffe 学习专栏收录该内容

9 篇文章 0 订阅

订阅专栏

参考博客：https://blog.csdn.net/auto1993/article/details/70941440
在上一篇博客中已经训练好了 mnist 识别手写数字的模型，这篇博客就利用 caffe 的 python 接口对得到的模型进行测试

from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import caffe
from skimage import io

imgsource = 'F:/python_work/mnist/images/001.png'
img = Image.open(imgsource)
plt.figure("image")
plt.imshow(img)
plt.show()

modefile = 'F:/caffe-master/examples/mnist/lenet.prototxt'
pretrain = 'F:/caffe-master/examples/mnist/lenet_iter_10000.caffemodel'
inputImage = caffe.io.load_image(imgsource,color=False)
net = caffe.Classifier(modefile,pretrain)
prediction = net.predict([inputImage],oversample=False)
caffe.set_mode_gpu()
print'predicted classes:',prediction[0].argmax()

运行结果
在这里插入图片描述