caffe实战二：mnist训练和测试

1.获取mnist训练和测试数据

cd data/mnist
sudo sh get_mnist.sh

2.转换数据成lmdb

sudo sh examples/mnist/create_mnist.sh

3.配置

因为不用GPU 所以修改examples/mnist/lenet_solver.prototxt文件,将solver_mode:GPU改为 solver_mode:CPU

4.训练

sudo sh ./examples/mnist/train_lenet.sh

最终得到lenet_iter_10000.caffemodel训练模型文件

5.测试

命令行方式 在examples/mnist下新建test_lenet.sh 内容为：

#!/usr/bin/env sh

build/tools/caffe test --model=examples/mnist/lenet_train_test.prototxt --weights=examples/mnist/lenet_iter_10000.caffemodel -iterations 100

回到caffe根目录执行：

sudo sh examples/mnist/test_lenet.sh

执行完毕图片：