手写字符识别是任何深度学习框架入门的首选,在caffe中实现mnist也非常的简单,大致步骤如下:
一 首先在caffe的根目录下打开终端,然后在终端输入:
./data/mnist/get_mnist.sh
输入以后终端就开始下载mnist手写字符识别的数据,数据包括四个文件,网速不好的朋友可以到我的百度云盘去下载,
云盘地址,链接: https://pan.baidu.com/s/1cZ1tZC 密码: swp5
二 mnist数据集准备好了以后,就可以开始制作caffe训练时所需要的lmdb格式的数据了,在终端输入:
./examples/mnist/create_mnist.sh
执行这个时,就开始制作lmdb数据了
制作完后会在/examples/mnist/目录下生成mnist_train_lmdb和mnist_test_lmdb两个文件夹,文件夹下分别存贮的是train和test时的lmdb数据,两个文件夹下均为data.mdb和lock.mdb两个文件。
三 数据准备好了以后,就开始训练模型了,在终端输入:
./examples/mnist/train_lenet.sh
这时候在命令行窗口就汇出现模型网络结构信息以及训练时的一些参数!!!慢慢等着训练完就可以了。
训练完后在./examples/mnist/下会生成caffemodel文件和solvoerstate文件,caffemodel文件是用来识别用的,solverstate 文件是继续训练网络用的