运行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