运行caffe自带的mnist实例教程

运行caffe自带的mnist实例教程
1、先进入caffe文件目录，(指令：cd ./caffe)，再用data/mnist下的get_mnist.sh下載MNIST数据集，代码如下：

cd ./caffe
sudo sh ./data/mnist/get_mnist.sh

2、转换格式，代码如下：

 ./examples/mnist/create_mnist.sh

3、训练超参数，有两种方式：

• （1）通过命令行执行训练，代码如下：（这是书上的例子，可以参考网址上的例子，在后面。不同之处：网址上的例子在此处新建一个文件夹保存训练的模型）

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

• 2）将以上训练的命令行代码写成训练脚本并命名为train_lenet.sh，放到mnist目录下。运行train_lenet.sh脚本进行训练。

./examples/mnist/train_lenet.sh

5、用训练好的模型对数据进行预测。

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