【caffe-windows】 caffe-master 之 mnist 超详细

本教程尽量详细,大多步骤都有图,如果运行出错,请先对照自己的文件是否和图上的一样,包括标点啊,空格啊,斜杠,反斜杠啊类的小细节。

第一步:   官网下载mnist数据 ,共4个文件,解压放到caffe-master\data\mnist下

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

第二步: 转换成caffe需要的数据格式,此处转换为 LMDB(cifar是转换成 LEVELDB,可对比参考如何实现的) ,转换格式需要用到caffe里的项目convert_mnist_data ,这个convert_mnist_data.exe程序是放在caffe-master\Build\x64\Release下,如果是用 debug生成的解决方案,则是在caffe-master\Build\x64\Debug下,这个在写 bat文件的时候要注意更改相应的路径。 

在caffe-master\data\mnist下新建一个文本文档,更改后缀为 .bat  在里面复制以下code: 

..\..\Build\x64\Release\convert_mnist_data.exe  ..\..\data\mnist\train-images.idx3-ubyte ..\..\data\mnist\train-labels.idx1-ubyte ..\..\examples\mnist\mnist_train_lmdb  

echo.  

..\..\Build\x64\Release\convert_mnist_data.exe  ..\..\data\mnist\t10k-images.idx3-ubyte  ..\..\data\mnist\t10k-labels.idx1-ubyte ..\..\examples\mnist\mnist_test_lmdb 

pause  

其实第一个路径就是 exe文件所在路径, 后面两个就是数据所在路径,最后一个是输出文件路径

保存后,双击运行

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

(PS:想转换成LEVELDB格式的话,只需要在.bat文件中把 mnist_train_lmdb改成mnist_train_leveldb ,有两处喔! )

第三步: 更改lenet_train_test.prototxt 和lenet_solver.prototxt文件,主要是改数据的路径以及数据的格式(默认LMDB,若在第二步转成了LEVELD的话,要改成LEVELDB )

首先 lenet_train_test.prototxt改的有

source: "../../examples/mnist/mnist_train_lmdb"

batch_size: 64

backend: LMDB

source:"../../examples/mnist/mnist_test_lmdb"

batch_size: 100

backend: LMDB

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

lenet_solver.prototxt改的有

net:"../../examples/mnist/lenet_train_test.prototxt"

snapshot_prefix:"../../examples/mnist/lenet"

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

若用CPU在最后一行改成 CPU

第四步: 开始训练,和前我博客里cifar10一样,有两种方法,这里只介绍第一种方法,用批处理文件的方法(想知道第二种方法的请查看之前博客)

在caffe-master\examples\mnist 下建立文本文档,后缀改为 .bat  复制以下code:

..\..\Build\x64\Release\caffe.exe train--solver=../../examples/mnist/lenet_solver.prototxt

Pause

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

保存,双击运行,弹出命令窗,等待训练结束,我的显卡是 GTX 960M ,只需要1分07秒就迭代了10000次,准确率 0.9915(用了一台i3的笔记本,cpu跑的,用了17分钟!)

如图:

【caffe-windows】 caffe-master 之 mnist 超详细

PS: 总结两个遇到问题:

1. 转换数据时候要确定转换的是 LMDB还是LEVELDB,这个是在一开始转换的时候设置的,一旦设置,在后面的train_test.prototxt文件中一定要对应的改过来,切记,切记。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zzddsw.html