Caffe学习(2) Windows 版本 运行mnist数据集

第一步 下载mnist数据

下载地址http://yann.lecun.com/exdb/mnist/

下载后解压到caffe-master\data\mnist文件夹中

第二步 转换数据格式

由于下载到的数据集为二进制文件，需要转换为特定格式才能被Caffe识别

在caffe-master根目录下创建creat_mnist.bat文件，并编辑如下指令

.\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中多了这两个文件夹

文件夹中的内容如下所示：

第三步 训练

找到\examples\mnist\lenet_solver.prototxt 文件，这个文件用来编辑训练过程中的各项参数，

打开并修改最后一行为CPU(此时我们只用CPU训练)

在caffe-master根目录下创建文件 train_mnist.bat,并编辑：

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

保存，双击运行，然后就开始训练啦。

训练过程中，在caffe-master\examples\mnist中会生成训练状态文件和参数文件

比如：

第四步 对训练好的网络进行测试

在caffe-master根目录下创建文件test_mnist.bat,并编辑

.\Build\x64\Release\caffe.exe test 

-model .\examples\mnist\lenet_train_test.prototxt

 -weights  .\examples\mnist\lenet_iter_2000.caffemodel 

-iterations 100

其中-weights后面的模型文件可以根据自己生成的模型文件来修改

保存文件，双击运行。