Caffe运行MNIST | example

参考教程测试一下caffe。

1.获取数据包。

cd $CAFFE_ROOT
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

官网上是这样的,caffe_root是根目录,看网上有人说必须在根目录下运行,否则会出错,具体没有验证。。但是windows平台下应该是没有wget的,需要自己下载一下。得到四个文件,测试与训练,样本与标签。

第二句是个坑,直接执行的话会提示找不到convert_mnist_data.bin(好像是这个)。这个环境还是linux下的,windows下编译出来的是exe,和caffe在一起。用法是

 convert_mnist_data [FLAGS] input_image_file input_label_file output_db_file

执行两次将mnist date转化为可用的lmdb格式的文件。并将新生成的2个文件mnist-train-lmdb 和 mnist-test-lmdb放于create_mnist.sh同目录下。

2.测试

参数文件用的是

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

mnist_test_lmdb mnist_train_lmdb 两个文件夹需要放在\examples\mnist。如果像我没用GPU,还需要在.prototxt里面更改solver_mode为 CPU。

感觉跑了一个半小时吧,终于跑完了。生成了四个文件……数据都训练好之后,接下来就是如何将模型应用到实际数据了(记录的博客):

./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel -gpu=0 

如果没有GPU则使用

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

看起来准确率很高。

Training LeNet on MNIST with Caffe