一、下载mnist数据集
下载地址:http://yann.lecun.com/exdb/mnist/我们需要下载的是如图4个压缩包:
![](https://img-blog.csdnimg.cn/20181126151749554.png)
![](https://img-blog.csdnimg.cn/20181126151951541.png)
二、处理数据
我们需要将原始数据处理成lmdb数据文件。在caffe-master下创建一个windows下的批处理文件create_mnist.bat
,将如下命令复制到文件中,然后双击create_mnist.bat
执行命令。 或者在cmd下进入caffe-master逐条执行如下命令。
.\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
完成后,我们在caffe-master\examples\mnist 下获得两个文件夹:
![](https://img-blog.csdnimg.cn/20181126152322496.png)
如果不成功,可能是文件夹位置不对,或者编译caffe时,convert_mnist_data的project没有编译成功。
三、修改配置文件
1、打开 caffe-master\examples\mnist\lenet_solver.prototxt 修改solver_mode为CPU。因为在上一篇博客中博主安装caffe配置的onlycpu。
2、打开 caffe-master\examples\mnist\lenet_train_test.prototxt 修改两处的source(指出训练和测试lmdb数据所在的位置),将 examples/mnist/mnist_train_lmdb
改为 ./examples/mnist/mnist_train_lmdb
,examples/mnist/mnist_test_lmdb
改为./examples/mnist/mnist_test_lmdb
。
四、训练和测试
在caffe-master下创建一个windows下的批处理文件train_mnist.bat,将如下命令复制到文件中,然后双击train_mnist.bat执行命令。 或者在cmd下进入caffe-master逐条执行如下命令。就会开始训练,训练过程中会输出一些信息。
.\Build\x64\Release\caffe.exe train --solver=.\examples\mnist\lenet_solver.prototxt
pause
![](https://img-blog.csdnimg.cn/20181126154034507.png)
输出Optimization Done.
训练结束。训练完成后在caffe-master\examples\mnist文件夹下会生成训练好caffemodel的模型。
参考资料
[1] https://blog.csdn.net/liuweiyuxiang/article/details/79532073
[2] https://blog.csdn.net/u011995719/article/details/54023085