Caffe实战：LeNet模型训练测试MNIST数据集

MNIST转成lmdb格式

1. 首先下载minist数据文件。此处我放的路径为E:\installpacket\ 编程软件\caffe 源码\caffe-master\caffe-master\examples\mnist\minist （注意：路径中中最好不要包含空格等字符，否则有可能会出现问题）

   

2. 在该文件夹下建立bat文件。内容如下：

   set DATA=E:\installpacket\编程软件\caffe源码\caffe-master\caffe-master\examples\mnist\minist

   set TOOLS=E:\installpacket\编程软件\caffe源码\caffe-master\caffe-master\Build\x64\Release

    

   REM set BACKEND=leveldb or lmdb

   set BACKEND=lmdb

    

   echo "Creating %BACKEND%..."

    

   rd /s /q "mnist_train_%BACKEND%"

   rd /s /q "mnist_test_%BACKEND%"

    

   "%TOOLS%\convert_mnist_data.exe" %DATA%\train-images-idx3-ubyte %DATA%\train-labels-idx1-ubyte mnist_train_%BACKEND% --backend=%BACKEND%

   "%TOOLS%\convert_mnist_data.exe" %DATA%\t10k-images-idx3-ubyte %DATA%\t10k-labels-idx1-ubyte mnist_test_%BACKEND% --backend=%BACKEND%

    

   echo "Done."

    

   Pause

    

   备注：REM为脚本文件的行注释，rd为删除的意思，/s 表示删除该目录下所有遍历的子目录和文件，/q 删除目录时不提示确认，直接删除.

    

    

    

3. 执行完会出现如下两个文件夹：

   

    

 

 

MNIST训练

 

1. window下打开cmd ，转移到，一定要转移到caffe-master 目录下。

   

2. 将mnist_test_lmdb和mnist_train_lmdb 文件夹进行移动。因为里面有很多程序默认这两个文件在移动后的那个文件夹中。

   移动前：

   

   移动后：

   

3.输入如下代码开始训练。

4.训练完成后，会多出如下几个文件。为训练好的权值文件。

 

 

 

测试

 

1.创建一个bat文件内容如下：

e:

cd \installpacket\编程软件\caffe源码\caffe-master\caffe-master

 

E:\installpacket\编程软件\caffe源码\caffe-master\caffe-master\Build\x64\Release\caffe.exe test -model=E:\installpacket\编程软件\caffe源码\caffe-master\caffe-master\examples\mnist\lenet_train_test.prototxt -weights=E:\installpacket\编程软件\caffe源码\caffe-master\caffe-master\examples\mnist\lenet_iter_10000.caffemodel

 

Pause

 

备注：得先转移到e:\installpacket\编程软件\caffe源码\caffe-master\caffe-master目录中，因为cpp中的许多路径为相对路径，是以此为基础的。

 

2.运行结果如下：