Mnist 图片测试

2018-12-09  本文已影响0人  菜囧爱学习

1、目的

在上述训练模型的基础上,测试自己手写的图片,完成手写图片的生成、图片转caffe输入格式文件生成以及模型测试。

2、手写图片、caffe输入格式转换

windows画图软件,属性设置图片像素28*28
图片黑底白字,
用matlab将图片转化为mnist二进制格式文件,
clc;
clear;
for i=1:10
imgOrigin=imread([num2str(i,'%d'),'.png']);
imgBinaryImg(i,:,:)= imgOrigin(:,:,1).'; 
end
imgMnist = fopen('test_img_ubyte', 'wb');  
magic = 2051;
numImages = i;
numRows = 28;
numCols = 28;  
fwrite(imgMnist, magic, 'int32', 0, 'ieee-be'); 
fwrite(imgMnist, numImages, 'int32', 0, 'ieee-be');
fwrite(imgMnist, numRows, 'int32', 0, 'ieee-be');  
fwrite(imgMnist, numCols, 'int32', 0, 'ieee-be');  
for i=1:10
imgBinaryImg1(numCols* numRows*(i-1)+1:numCols* numRows*i)= reshape(imgBinaryImg(i,:,:), 1, numCols* numRows); 
end
fwrite(imgMnist, imgBinaryImg1, 'unsigned char');  
fclose(imgMnist);  
flabel = fopen('test_lable_ubyte', 'wb');  
magic = 2049;
numLabels = i;
labels = [5 6 1 4 7 0 9 1 3 5];
fwrite(flabel, magic, 'int32', 0, 'ieee-be');  
fwrite(flabel, numLabels, 'int32', 0, 'ieee-be');  
fwrite(flabel, labels, 'unsigned char');  
fclose(flabel);  
运行后产生文件

将其放于\caffe\data\mnist文件夹。
git下运行/mnist/create_mnist_byourownselves.sh ,产生caffe读入文件

#!/usr/bin/env sh
# This script converts the mnist data into lmdb/leveldb format,
# depending on the value assigned to $BACKEND.
set -e
EXAMPLE=examples/mnist
DATA=data/mnist
BUILD=Build/x64/Release
BACKEND="lmdb"
echo "Creating ${BACKEND}..."
rm -rf $EXAMPLE/mnist_train_${BACKEND}
rm -rf $EXAMPLE/mnist_test_${BACKEND}
$BUILD/convert_mnist_data.exe $DATA/train-images-idx3-ubyte \
  $DATA/train-labels-idx1-ubyte $EXAMPLE/mnist_train_${BACKEND} --backend=${BACKEND}
$BUILD/convert_mnist_data.exe $DATA/test_img_ubyte \
  $DATA/test_lable_ubyte $EXAMPLE/mnist_test_${BACKEND} --backend=${BACKEND}
echo "Done."
产生测试图片为10张,修改lenet_train_test.prototxt

batch_size为10,
git bash下运行,

./Build/x64/Release/caffe.exe test -model examples/mnist/lenet_train_test_byourself.prototxt  -weights examples/mnist/lenet_iter_10000.caffemodel  -iterations 1
测试结果为:

所有代码以及资料地址:

https://github.com/Alix1992/Caffe

参考博客:

https://blog.csdn.net/woyaopojie1990/article/details/50820526`
上一篇下一篇

猜你喜欢

热点阅读