windows下使用caffe测试mnist数据集

在win10机子上装了caffe，感谢大神们的帖子，要入坑caffe-windows的朋友们看这里，还有这里，安装下来基本没什么问题。

好了，本博文写一下使用caffe测试mnist数据集的步骤。

1. 下载mnist数据集。

不太看得懂get_mnist.ps1文件，并且运行无效，所以选择直接从mnist官网下载数据集。下载后解压，从解压后的文件夹提取出四个文件，放在caffe根目录下<caffe-root>\data\mnist下，例如E:\caffe-windows\data\mnist，否则后面运行报错找不到文件。

2. 创建lmdb文件。

也有的人会用leveldb，这里选择使用lmdb存储。在目录E:\caffe-windows\examples\mnist下，运行create_mnist.ps1文件，博主发现闪一下就退出了，抓屏后发现是找不到文件convert_mnist_data.exe，用编辑器打开，发现路径错误，看下图，这是修改过的，使用build_win.exe编译caffe时，build文件夹会建在scripts目录下，找到convert_mnist_data.exe的路径，修改create_mnist.ps1文件。可以看到$DATA后的数据，正是1中提取出的文件。

 

再次运行，在examples\mnist下生成两个lmdb文件夹。如图：

 

 

3. 修改参数

examples\mnist目录下lenet_solver.prototxt文件默认使用GPU训练，根据自己情况使用CPU或者GPU。并将文件中的路径改为绝对路径，lenet文件夹是我自己创建的，貌似不能自动创建？

 

 

同样，examples\mnist目录下lenet_train_test.prototxt文件，修改这里，绝对路径：

 

 

4. 开始训练。

改完参数了，创建bat文件训练。这里也可以使用examples\mnist目录下提供的train_lenet.ps1，运行出错，还不如用bat文件简简单单两句话

E:\caffe-windows\scripts\build\tools\Release\caffe.exe  train  --solver=E:\caffe-windows\examples\mnist\lenet_solver.prototxt
pause

 

运行结果如图：

 

 

Ok,done!