- LeNet网络模型描述文件(caffe-windows\examples\mnist\lenet_train_test.prototxt)
描述卷积神经网络结构
name: "LeNet" #网络的名字"LeNet" layer { #定义一个层(层从数据来源,传入数据,转化为LMDB,输出data和label,64张一批次,在训练时有效) name: "mnist" #层的名字"mnist" type: "Data" #层的类型"Data",表明数据来源于LevelDB或LMDB。另外数据的来源还可能是来自内存,HDF5,图片等 top: "data" #top输出,输出data top: "label" #输出label include { phase: TRAIN #该层只在TRAIN训练的时候有效 } transform_param { #数据的预处理 scale: 0.00390625 #1/256,将输入的数据0-255(像素为8位,255)归一化到0-1之间 } data_param { source: "E:/Caffe-windows/caffe-windows/examples/mnist/lmdb/train_lmdb" #数据来源 batch_size: 64 #每个批次处理64张图片 backend: LMDB #转化为LMDB的数据格式 } }
layer { #定义一个层,(层从数据来源,传入数据,转化为LMDB,输出data和label,100张一批次,在测试时有效) name: "mnist" #层的名字"mnist" type: "Data" #层的类型"Data",表明数据来源于LevelDB或LMDB top: "data" #输出dada top: "label" #输出label include { phase: TEST #该层只在TEST测试的时候有效 } transform_param { #数据的预处理 scale: 0.00390625 #1/256,将输入的数据0-255归一化到0-1之间 } data_param { source: |