Caffe 学习系列（三）：训练mnist数据集

#Caffe 学习系列（三）：训练mnist数据集

参考这篇

Caffe入门：LeNet模型介绍与网络训练测试例子

1、获取mnist数据集

cd ~/caffe
sduo ./data/mnist/get_mnist.sh

2、数据集格式转换

Caffe只识别 lmdb 格式的数据集，因此将二进制数据库文件转换成lmdb数据库格式

sudo ./examples/mnist/create_mnist.sh

执行之后，在./examples/mnist文件夹下生成两个文件夹，一个是mnist_train_lmdb 和 mnist_test_lmdb，分别存放了训练和测试数据

3、训练lenet网络

如果只是用CPU训练的话，需要先在 lenet_solver.prototxt 文件中修改训练模式为CPU，使用gedit打开.prototxt文件

sudo gedit ./examples/mnist/lenet_solver.prototxt

修改文件最后一行的 solver_mode的配置为 CPU

# solver mode: CPU or GPU
solver_mode: CPU

运行以下指令执行训练：

sudo ./examples/mnist/train_lenet.sh

训练结果的分类文件 lenet_iter_10000.caffemodel 文件保存在./examples/mnist路径下。