Caffe代码初识

caffe中prototxt文件定义
一、定义输入
参数batch_size,num_channel_s,input_height,input_width
batch_size:每次训练样本个数
num_channel:表示输入图片的通道数
input_height:表示在单个图片上,垂直方向每次移动的像素

input_width:表示单个图片上,水品方向每次移动的像素
二定义网络层
定义网络层,以卷积层为例:    
layer {
  name: "conv1_1"
  type: "Convolution"
  bottom: "data"
  top: "conv1_1"
  convolution_param {
    num_output: 64
    kernel_size: 3
  }
}

name为该层的名字,可以自己定义.type表示该层的特性,如relu,type=ReLU,pooling,type=Poolin,这里为卷积层,因此type=Convolution,bottom表述该层的输入,这里输入为input data. top表示该层的输出output data,*_param定义该层的参数信息,如

convolution_param {    
<code>num_output: 64
kernel_size: 3
</code>

}
表示该卷积层卷积核大小为3×3,output channels 为64,表示这层卷积层有64个卷积核.

猜你喜欢

转载自blog.csdn.net/xidaoliang/article/details/79738378