caffe之MNIST基础详解

cd $CAFFE_ROOT
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

get_mnist.sh:下载mnist的数据集。

create_mnist.sh是将下载好的mnist数据集转化为lmdb格式，新版本的caffe中默认将MNIST数据集转换为lmdb格式的文件。当然，你也可以根据需要用它生成leveldb格式的文件，只需要修改BACKEND=“leveldb”即可

各层详解：

打开lenet_train_test.prototxt查看详细信息

数据层定义如下图，需要注意的是source和backend，batch_size也可以根据需要自己调整。其中，参数scale用于对输入的像素进行缩放，将所有的像素值缩放到[0,1]区间内。0.00390625=1/256。最后该层会生成两个blob，一个是data，一个是label。

name: "LeNet"
layer {
#输入层的名字为mnist
  name: "mnist"
#输入的类型为DATA
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    scale: 0.00390625
  }
#数据的参数
  data_param {
#从mnist_train_lmdb中读入数据
    source: "examples/mnist/mnist_train_lmdb"
#一次处理64条数据
    batch_size: 64
    backend: LMDB
  }
}