window caffe 下训练mnist及测试

来源:互联网 发布:ubuntu没有中文输入法 编辑:程序博客网 时间:2024/05/22 14:29

训练
首先下载官网训练、测试文件:官网地址http://yann.lecun.com/exdb/mnist/ 解压到自己建立的文件夹下(文件夹名为英文)。
这里写图片描述
在该文件夹下保存一个.bat文件,文件内容为

E:\caffe-windows\Build\x64\Release\convert_mnist_data.ext train-images.idx3-ubyte train-labels-idx1-ubyte F:\mnist\mnist_train_lmdbecho.  E:\caffe-windows\Build\x64\Release\convert_mnist_data.exe t10k-images.idx3-ubyte t10k-labels.idx1-ubyte F:\mnist\mnist_test_lmdb pause 

(路径为绝对路径)在F:\mnist\下生成两个文件夹

这里写图片描述

在examples->mnist 中找到文件lenet_solver.prototxt与lenet_train_test.prototxt 打开。
将lenet_solver.prototxt 最后一行solver_mode:GPU 改为CPU,保存。
将lenet_train_test.prototxt中source路径修改为绝对路径,如图
这里写图片描述
保存文件。
在samples/mnist 文件夹下建.bat文件。文件内容为

E:\caffe-windows\Build\x64\Release\caffe.exe train -solver=E:\caffe-windows\examples\mnist\lenet_train_test.prototxtpause

运行.bat文件开始训练
这里写图片描述

最后产生准确率0.9915 loss=0.0285664
产生四个文件
这里写图片描述
训练成功

测试
在任意文件夹下建立一个.bat文件,文件内容为

E:\caffe-windows\Build\x64\Release\caffe.exe test -model=E:\caffe-windows\exmaples\mnist\lenet_train_test.prototxt  -weight=F:\mnist\lenet_iter_10000.caffemodel pause 

单击运行.bat文件,结果如图
这里写图片描述

准确率 0.9874 loss:0.0429096

测试完毕

1 0
原创粉丝点击