Caffe学习笔记1

MNIST数据集

运行caffe框架中的脚本下载MNIST数据集

cd data/mnist

./get_mnist.sh

用tree查看MNIST数据集结构

sudo apt-get install tree

tree

将二进制原始数据文件转换为LEVELDB或LMDB才能被Caffe识别

转换为LMDB格式

需要在根目录下运行否则找不到

./examples/mnist/create_mnist.sh

查看生成的结果：data.mdb和lock.mdb

ls -l examples/mnist/mnist_train_lmdb/

ls -l examples/mnist/mnist_test_lmdb/

vi examples/mnist/lenet_solver.prototxt

修改求解模式GPU为CPU

solver_mode:CPU

caffe根目录下运行

./examples/mnist/train_lenet.sh

输出结果可以看到训练的模型权值保存在examples/mnist/lenet_iter_10000

训练状态保存在examples/mnist/lenet_iter_10000.solverstate

用训练好的LeNet-5模型权值对测试数据集进行预测

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

./build/tools/caffe.bin test                                                     表示只做预测

examples/mnist/lenet_train_test.prototxt                             指定模型描述文本文件

-weights examples/mnist/lenet_iter_10000.caffemodel       指定模型预先训练好的权值文件

-iterations 100                                                                      指定测试迭代次数