caffe-master 之 mnist

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


第一步:   官网下载mnist数据 http://yann.lecun.com/exdb/mnist/,共4个文件,解压放到caffe-master\data\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  

扫描二维码关注公众号,回复: 3565200 查看本文章

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文件所在路径, 后面两个就是数据所在路径,最后一个是输出文件路径

保存后,双击运行


如图:



(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


如图:


 


lenet_solver.prototxt改的有

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

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


如图:


 

若用CPU在最后一行改成 CPU

 

 

第四步: 开始训练,和之前我博客里cifar10一样,有两种方法,这里只介绍第一种方法,用批处理文件的方法(想知道第二种方法的请查看之前博客http://blog.csdn.net/u011995719/article/details/53998331

 

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

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

Pause

 

如图:


 

 

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

如图:


 


PS: 总结两个遇到问题:

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

2.Debug 和 Release    ,这个是在VS2013中选择的,具体有什么区别我不太清楚(百度了,说对使用问题不大),但是我们在调用例如 convert_mnist_data.exe 、compute_image_mean.exe等这些 应用程序时,要选择路径,一定要看清楚你生成的是在Debug文件夹(caffe-master\Build\x64\Debug)里还是在Release(caffe-master\Build\x64\Release)里,如果你生成的是在Debug里,而调用的时候是用路径:. Build\x64\Release 的话,就会出现如下错误:



当然,可以在VS2013中分别 选择 Debug和Release 进行生成,这样就不用担心第二个问题啦。

猜你喜欢

转载自blog.csdn.net/yangpan011/article/details/79686609