在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!