- 获取mnist数据
官网下载:http://yann.lecun.com/exdb/mnist/
解压后存入.\data\mnist (. 是caffe根目录)
- 转换成lmdb格式数据
在.\examples\mnist\下建立文件create_mnist.bat,内容如下:
echo "create train data"
.\scripts\build\examples\mnist\Release\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb
echo "create test data"
.\scripts\build\examples\mnist\Release\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte .\data\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb
pause
第一个是convert_mnist_data.exe所在路径,第二个是第一步中获取到数据的路径,最后一个是输出数据。输出的mnist_train_lmdb和mnist_test_lmdb文件夹中各有data.mdb和lock.mdb两个文件。
保存后双击运行。
- 修改.\examples\mnist\lenet_solver.prototxt,最后一行
solver_mode:CPU
- caffe根目录建立文件train_mnist.bat,内容如下:
.\scripts\build\tools\Release\caffe.exe train --solver=.\examples\mnist\lenet_solver.prototxt
pause
保存后双击运行。