caffe之MNIST基础详解

来源：互联网发布：政府财政支出数据编辑：程序博客网时间：2024/05/01 15:06

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  }}

0 0