(1)去官网http://yann.lecun.com/exdb/mnist/下载mnist数据集。
解压放到D:\caffe-master\data\mnist目录下;
(2)在caffe根目录下,新建一个mycreate_mnist.bat,里面写入如下的脚本。
.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb
echo.
.\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte .\data\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb
pause
然后双击该脚本运行,即可在E:\caffe\examples\mnist下面生成两个文件夹,及相应的lmdb数据文件。
注意:如果该文件夹存在或文件夹里的文件存在,会出现错误,需要删除后再生成。
(3)修改E:\caffe\examples\mnist\lenet_solver.prototxt,将最后一行改为solver_mode:CPU,
修改E:\caffe\examples\mnist\lenet_train_test.prototxt,讲train和test数据地址修改成lmdb文件夹
data_param {
source: "D:/caffe-master/examples/mnist/mnist_train_lmdb"
batch_size: 64
backend: LMDB
}
data_param {
source: "D:/caffe-master/examples/mnist/mnist_test_lmdb"
batch_size: 100
backend: LMDB
}
(4)在caffe根目录下,cmd中运行如下命名
就会开始训练,训练完毕后会得到相应的准确率和损失率。
i3破电脑上运行的时间是:11分钟