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文件,在这个文件夹内需要修改测试数据的路径,如果在训练时按照我的方法修改过则无需再改,直接到下一步。 

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


然后在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%以上。



  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值