Windows下caffe之测试mnist

一、参考资料

悠望南山-CAFFE学习笔记(二)Caffe_Example之测试mnist

二、训练所需要用到的文件

1.mnist_test_lmdb  (测试数据集,里面包括data.mdb和lock.mdb)

2.lenet_train_test.prototxt  (训练和测试所用到的net模型数据,包括各种layer,如卷积层conv,采样层也即池化层pool,全连接层ip,非线性层Relu,损失层loss。数据以blob形式在各个层中传递)

3.test_lenet.bat(此文件中存放进行测试的命令,是需要自己创建的文件,文中会提及)

4.lenet_iter_10000.caffemodel(训练后生成的权重文件)

三、准备工作

测试依然在训练mnist数据集时创建的文件夹下进行。即在caffe\build\x64\Debug或者caffe\build\x64\Release文件夹下,与你的caffe.exe放在同一个文件夹下即可。然后将caffe\examples\mnist文件夹下的两个lmdb数据文件夹拷贝到刚才新建的文件夹下,同时在mnist文件夹下寻找其他三个之前提到的文件并拷贝到新建的文件夹下。

首先打开lenet_train_test.prototxt文件,在这个文件夹内需要修改测试数据的路径,如果在训练时按照我的方法修改过则无需再改,直接到下一步。 

没有修改的按照如下截图所示内容进行修改,前提是你已经按照前文所说的方式创建好了文件夹,并且将所需的数据移动到了该文件夹下。

Windows下caffe之测试mnist_第1张图片

然后在Release文件夹下(也可能是Debug文件夹,看个人情况而定)新建txt格式文件,重命名为test_lenet.bat,并打开进行编辑,输入下面这段代码:

#!/usr/bin/env sh

C:\Users\MaMQ\Desktop\caffe-master\Build\x64\Release\caffe.exe test -model=C:\Users\MaMQ\Desktop\caffe-master\Build\x64\Release\example\lenet_train_test.prototxt -weights=C:\Users\MaMQ\Desktop\caffe-master\Build\x64\Release\example\lenet_iter_10000.caffemodel

pause
其中,-model是调用网络模型文件,-weights是调用训练时生成的权重文件。pause是为了在运行结束后窗口不立即关闭,而是等待我们从键盘输入之后关闭,方便我们查看log信息。

至此,准备工作完毕。

四、开始测试

双击test_test.bat文件,然后会弹出一个运行窗口,等待几分钟就可以看到结果了,发现测试的准确率基本都在98%以上。Windows下caffe之测试mnist_第2张图片



你可能感兴趣的:(caffe)