文章目录
LeNet-5模型描述
本节例程中的LeNet-5模型与原版稍有不同(例如,将Sigmoid激活函数改为ReLU),其描述文件在Caffe目录下的examples/mnist/lenet_train_test.prototxt
中,内容及代码注释如下:
name: "LeNet" //网络(Net)的名称为LeNet
layer {
//定义一个层(Layer)
name: "mnist" //层的名称为mnist
type: "Data" //层的类型为数据层
top: "data" //层的输出blob有两个:data和label
top: "label"
include {
phase: TRAIN //该层参数只在训练阶段有效
}
transform_param {
scale: 0.00390625 //数据变换使用的数据缩放因子
}
data_param {
//数据层参数
source: "examples/mnist/mnist_train_lmdb" //LMDB的路径
batch_size: 64 //批量数目,一次读取64张图
backend: LMDB //数据格式为LMDB
}
}
layer {
//一个新数据层,名字也叫mnist,输出blob也是data和label,但是这里定义的参数只在分类阶段有效
name: "mnist"
type: "Data"
top: "data"
top: "label"
include {
phase: TEST
}
transform_param {
scale: 0.0039062