文章目录
1. mnist 数据集下载2. mnist 数据集转换3. 修改网络结构文件 lenet_train_test.prototxt 和 网络求解文件 lenet_solver.prototxt4. 开始训练上一篇博客写了在 win7 下配置 caffe,本篇博客就记录一下如何训练 mnist 数据集。
1. mnist 数据集下载
/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 的测试集和自己手写的数字进行测试。