第一步:
从mnist官网下载的四个文件并解压缩,并存放在路径C:\caffe-windows\examples\mnist下存在我在mnist下新建了一个mnist文件夹
并在examples\mnist文件夹下新建bat文件创建训练集create_minist_lmdb.bat完成数据格式的转换,内容如下:
C:\caffe-windows\Build\x64\Release\convert_mnist_data.exe ./mnist/train-images.idx3-ubyte ./mnist/train-labels.idx1-ubyte ./mnist_train_lmdb
C:\caffe-windows\Build\x64\Release\convert_mnist_data.exe ./mnist/t10k-images.idx3-ubyte ./mnist/t10k-labels.idx1-ubyte ./mnist_test_lmdb
pause
运行成功后,生成两个新文件夹及相关数据。
注意:如果重复生成会报错。
第二部:
C:\caffe-windows\examples\mnist\lenet_solver.prototxt 参数修改
仅使用CPU
第三步:
在caffe根目录下生成run.bat,内容如下:
Build\x64\Release\caffe.exe train --solver=examples/mnist/lenet_solver.prototxt
pause
运行需要一些时间,结果如下: