как проверить 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

Почему вы не используете питон обертку класса 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.]]