(caffe入门)windows caffe 之 mnist 训练

文章目录

1. mnist 数据集下载
2. mnist 数据集转换
3. 修改网络结构文件 lenet_train_test.prototxt 和网络求解文件 lenet_solver.prototxt
4. 开始训练

上一篇博客写了在 win7 下配置 caffe，本篇博客就记录一下如何训练 mnist 数据集。

1. mnist 数据集下载

http://yann.lecun.com/exdb/mnist/

解压并将其中的文件放到目录E:\caffe-master\data\mnist下

2. mnist 数据集转换

下载得到的数据集不能直接使用，需要转换成 caffe 需要的数据格式，此处转换成 leveldb，转换格式用到的工具是 convert_mnist_data.exe

在 E:\caffe-master\data\mnist 目录下新建 convert_leveldb_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_lmdbecho...\..\Build\x64\Release\convert_mnist_data.exe ..\..\data\mnist\t10k-images.idx3-ubyte ..\..\data\mnist\t10k-labels.idx1-ubyte ..\..\examples\mnist\mnist_test_lmdbpause

这里转换为 lmdb 格式

内容依次为转换工具 .exe 路径，数据集所在路径，最后是转换后的数据存放路径

保存，双击运行，这是转换为 leveldb 的截图，lmdb 类似

这时，你会发现 example 目录下多了两个文件夹，如图

3. 修改网络结构文件 lenet_train_test.prototxt 和网络求解文件 lenet_solver.prototxt

lenet_train_test.prototxt 文件修改如下：

lenet_solver.prototxt 文件作如下修改：

4. 开始训练

在 E:\caffe-master\examples\mnist 目录下新建 caffe_mnist_train.bat 文件，写入代码

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

双击运行

报错显示驱动版本与运行不符，老泪纵横啊，编译 caffe 的时候要 cuda7.5，我卸载的 cuda9.0 又需要重新装回去。
装好 cuda9.0 后再次运行，在 GPU 下跑了两三分钟，如图

第一个 caffe 模型终于训练成功啦！
输出从左到右一次为日期时间进程号文件名行号迭代次数损失值
训练结束后，会在 examples/mnist 目录下生成训练得到的模型，可以用这模型对 mnist 的测试集和自己手写的数字进行测试。