LeNet-5模型prototxt文件描述及各层含义、注释

本文介绍了LeNet-5模型的prototxt文件内容及其网络结构,模型采用了ReLU激活函数,并从数据源读取预处理后的mnist数据。CNN结构包含卷积层和下采样层的交替,以及两个全连接层,用于图像分类。理解LeNet有助于深入学习更大规模的CNN模型设计。
摘要由CSDN通过智能技术生成

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
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值