Caffe MNIST 入门记录

Deep-Learning 2017-09

记录Caffe训练模型的入门教程-MNIST

准备材料

• 搭建Caffe环境请参考 Ubuntu 16.0.4 配置Caffe 图文记录

网络不好的情况如下处理：

• THE MNIST DATABASE of handwritten digits 请下载训练模型的数据集（如果网络不好的话，这里可以手动下载）
• 然后将上述数据集解压后放入caffe/data/minist目录下

t10k-images-idx3-ubyte
t10k-labels-idx1-ubyte
train-images-idx3-ubyte
train-labels-idx1-ubyte

网络好的话直接通过下载脚步处理即可。

sudo sh data/mnist/get_mnist.sh

转换为Caffe适用的数据LMDB格式

通过脚本转换格式：

sudo sh examples/mnist/create_mnist.sh

生成的数据如图所示：

配置

接下来如果Caffe配置了GPU的话，不需要再配置了；
如果没有GPU配置的话，需要手动修改配置文件。

sudo vi examples/mnist/lenet_solver.prototxt

可以修改max_iter最大迭代次数，可以修改solver_mode为GPU还是CPU，这里默认GPU，如图所示。

训练

配置完成以后，可以直接训练了。

sudo time sh examples/mnist/train_lenet.sh

训练结束以后，会自动生成.prototxt网络配置文件和.caffmodel模型文件

最后输出模型的精度和loss如下：