一、下载mnist数据集

下载地址:http://yann.lecun.com/exdb/mnist/我们需要下载的是如图4个压缩包:

Caffe安装测试——mnist训练
将下载的数据集放在caffe-master\data\mnist下,并解压。
Caffe安装测试——mnist训练

二、处理数据

我们需要将原始数据处理成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 下获得两个文件夹:

Caffe安装测试——mnist训练

如果不成功,可能是文件夹位置不对,或者编译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_lmdbexamples/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 
Caffe安装测试——mnist训练

输出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

分类:

技术点:

相关文章: