windows下使用caffe测试mnist数据集

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

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

1. 下载mnist数据集。

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

2. 创建lmdb文件。

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

 

再次运行,在examplesmnist下生成两个lmdb文件夹。如图:

 

3. 修改参数

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

 

同样,examplesmnist目录下lenet_train_test.prototxt文件,修改这里,绝对路径:

 

4. 开始训练。

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

E:caffe-windowsscriptsuild	oolsReleasecaffe.exe  train  --solver=E:caffe-windowsexamplesmnistlenet_solver.prototxt
pause

运行结果如图:

 

Ok,done!

原文地址:https://www.cnblogs.com/peony-jing/p/8617532.html