【Python学习】 caffe-master 之 mnist 详解

作者：佛山小程序员
原文：https://blog.csdn.net/weixin_44192923/article/details/86563313

第一步：

官网下载mnist数据http://yann.lecun.com/exdb/mnist/，共4个文件，解压放到caffe-master\data\mnist下

最近开始整理python的资料，会陆续放到博客中存档。找了几个qq群，其中有一个群78486745。后面就没怎么加群了，还是需要看官方文档为主

第二步：

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

保存后，双击运行

（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: 64backend: LMDBsource:"../../examples/mnist/mnist_test_lmdb"batch_size: 100backend: 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 进行生成，这样就不用担心第二个问题啦。

欢迎加群讨论