如何在我自己的數據集圖像上測試mnist

Question

我試圖用我自己的數字圖像數據集來測試mnist。
我寫了一個python腳本，但它給出了一個錯誤。錯誤在第16行代碼中。其實我無法發送圖像進行測試。給我一些建議。提前致謝。

import numpy as np 
import sys 
import caffe 
import matplotlib.pyplot as plt 
import os 

caffe_root = '../caffe-master/' 
MODEL_FILE = './examples/mnist/lenet.prototxt' 
PRETRAINED = './examples/mnist/lenet_iter_10000.caffemodel' 
IMAGE_FILE = '/home/hemant/OpenCVProject/grey/img001-00001.png'#image path 

input_image = caffe.io.load_image(IMAGE_FILE) 

net = caffe.Net(MODEL_FILE, PRETRAINED,caffe.TEST) 
caffe.set_mode_cpu() 
out = net.forward([input_image]) 
print out['prob'] 

Answer 1

爲什麼不使用python包裝類Classifier？

net = caffe.Classifier(MODEL_FILE, PRETRAINED) 
net.predict([input_image], oversmaple=False) 

我不是100％肯定，但我認爲LeNeT模型預計灰度圖像，你可能需要讀取圖像

input_image = caffe.io.load_image(IMAGE_FILE, color=False) 

Answer 2

import caffe 
import os 

model_file = '../examples/mnist/lenet.prototxt' 
pretrained_file = '../examples/mnist/lenet_iter_10000.caffemodel' 
net = caffe.Classifier(model_file, pretrained_file, image_dims=(28, 28), raw_scale=255) 
score = net.predict([caffe.io.load_image('img/1.bmp', color=False)], oversample=False) 
print score 

這對我的代碼工作，輸出是這樣的：

... 
[[ 0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]]