ubuntu16.04+caffe训练mnist数据集

1.   caffe-master文件夹权限修改

下载的caffe源码编译的caffe-master文件夹貌似没有写入权限，输入以下命令修改：

sudo chmod -R 777 ~/caffe-master/

2.   下载mnist数据库

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

caffe中的./data/mnist/get_mnist.sh 文件实现了下载mnist数据库的功能，文件的内容如下：

执行之后，在./data/mnist文件夹下生成4个文件，分别是测试、训练数据库和测试、训练标签： 

3.   二进制数据库文件转换成lmdb数据库格式 

sudo ./examples/mnist/create_mnist.sh

create_mnist.sh文件用于把数据库转换成lmdb格式，内容如下：

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

4.   训练lenet网络

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

sudo gedit ./examples/mnist/lenet_solver.prototxt

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

以下指令执行训练：

sudo ./examples/mnist/train_lenet.sh

训练的最终结果：

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

来源：https://www.cnblogs.com/mtcnn/p/9411831.html