【caffe for Python】classifier.py分类填坑

心血来潮，利用classifier.py做MNIST手写数据分类时出现两个坑

使用的代码出处http://blog.csdn.net/longji/article/details/70242230

分类代码

import os
import sys
import numpy as np
import matplotlib.pyplot as plt
caffe_root = '******/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe
MODEL_FILE='******/caffe/examples/mnist/lenet.prototxt'
PRETRAINED = '******/caffe/examples/mnist/lenet_iter_10000.caffemodel'
IMAGE_FILE='******/MNIST_test/train_0.bmp' # 7
input_image = caffe.io.load_image(IMAGE_FILE, color=False)
net = caffe.Classifier(MODEL_FILE, PRETRAINED)

# 预测图片分类，没有crop时，oversample过采样为false
prediction = net.predict([input_image], oversample = False)
caffe.set_mode_cpu()
# 打印分类结果
print('predicted class:', prediction[0].argmax())

第一坑

File "******\python\caffe\classifier.py", line 30, in __init__
	in_ = self.inputs[0]

IndexError: list index out of range

报错位置：classifier.py 第30行 在初始化函数中
索引错误：self.inputs的索引超出范围

print self.inputs   #输出为[]空列表

在网上找了老半天的毫无所获，直到看到了这个：
https://stackoverflow.com/questions/34651620/pycaffe-index-out-of-range-when-using-original-train-prototxt-as-the-pretrained
原因是说我的MODEL_FILE没有外部输入，应该使用deploy.prototxt部署文件，更改完后

print self.inputs   #输出为["data"]

第一坑填完

第二坑

File "******\python\caffe\classifier.py", line 85, in predict
    input_ = input_[:, crop[0]:crop[2], crop[1]:crop[3], :]

TypeError: slice indices must be integers or None or have an __index__ method

报错位置：classifier.py 第85行 在预测函数中
类型错误：切片索引必须使用整形或None或有__index__方法

print crop     #[  0.   0.  28.  28.]
print crop.dtype    #float64

所以改起来也很简单，在85行上加一个类型转换就好

crop=crop.astype(np.int32)
input_ = input_[:, crop[0]:crop[2], crop[1]:crop[3], :]

第二坑填完