python调用caffe模型_Caffe(二）使用Python运用caffemodel进行mnist数据集预测

Problem Solve

出错截图

出现以上截图，经百度之后原因为Ubuntu``protobuff的版本较高，导致出错，然而并没有找到解决办法。

但随即又出现'int object has no attribute '__values问题，百度Google易一通之后，依然无解。然后在Caffe Github的Issue模块中找到这个问题，原来是最新版本draw.py的Bug，下载caffe-rc4.zip替换之后成功解决问题。

转换mnist ubyte3到图像

代码如下

# -*- coding:utf-8 -*-

import struct

import numpy as np

import matplotlib.pyplot as plt

# import Image

from PIL import Image # import Image seems not work.

#二进制的形式读入

filename='train-images-idx3-ubyte'

binfile=open(filename,'rb')

buf=binfile.read()

#大端法读入4个unsigned int32

#struct用法参见网站 http://www.cnblogs.com/gala/archive/2011/09/22/2184801.html

index=0

magic,numImages,numRows,numColumns=struct.unpack_from