layer是建模和计算的基本单元。
caffe的目录包含各种state-of-the-art model的layers。
为了创建一个caffe model,我们需要定义模型架构在一个protocol buffer定义文件中(prototxt)。caffe的layer和它们的参数被定义在caffe.proto中。
Vision Layers:
头文件./include/caffe/vision_layers.hpp
vision layers通常取图像为输入,产生其他图像作为输出。实际中典型的图像可能只有一个颜色通道(c = 1),例如在一个灰度图像中,或者三个通道(c = 3),在一个RGB图像中。但是这里,一个图像的显著特征是它的空间结构,通常一个图像有高度h > 1,宽度w > 1。这个2D几何图形自然导致了如何处理输入。特别地,大多数vision layers通过对输入的一些区域应用一个特殊的操作,产生相应的输出。对比来看,其他layers(少数例外)忽略输入的空间结构,将输入视为一个大的向量,向量维度为chw。
Convolution layer:
- layer类型:Convolution
- CPU实现:.
/src/caffe/layers/conv_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/conv_layer.cu
- 参数(
ConvolutionParameter convolution_param
)- 必须要求的
- num_output(
c_o
): 滤波器数量 kernel_size
(orkernel_h
andkernel_w
): 每个滤波器的高和宽
- num_output(
- 强烈推荐的
- weight_filter [default
type: 'constant' value: 0
]
- weight_filter [default
- 可选的
bias_term
[defaulttrue
]: 是否学习和应用一组biase到滤波器输出pad
(orpad_h
andpad_w
) [default 0]: 指定在输入图像的每个边隐含添加的像素数目stride
(orstride_h
andstride_w
) [default 1]: 指定应用滤波器到图像时滤波器的间隔group
(g) [default 1]: 如果 g > 1,我们限制每个滤波器连接到输入的子集。特别地,输入和输出通道被分为g组,第i组输出仅仅连接到第i组输入。
- 必须要求的
- 输入: n * c_i * h_i * w_i
- 输出:n * c_o * h_o * w_o,其中h_o = (h_i + 2 * pad_h - kernel_h) / stride_h + 1,w_o可得类似结果。
- 例子:
layer { name: "conv1" type: "Convolution" bottom: "data" top: "conv1" # learning rate and decay multipliers for the filters param { lr_mult: 1 decay_mult: 1 } # learning rate and decay multipliers for the biases param { lr_mult: 2 decay_mult: 0 } convolution_param { num_output: 96 # learn 96 filters kernel_size: 11 # each filter is 11x11 stride: 4 # step 4 pixels between each filter application weight_filler { type: "gaussian" # initialize the filters from a Gaussian std: 0.01 # distribution with stdev 0.01 (default mean: 0) } bias_filler { type: "constant" # initialize the biases to zero (0) value: 0 } } }
Convolution layer卷积输入图像和一组可学习的滤波器,每个滤波器对应地产生输出图像的一个feature map。
Pooling layers:
- layer类型:Pooling
- CPU实现:.
/src/caffe/layers/pooling_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/pooling_layer.cu
- 参数(
PoolingParameter pooling_param
)- 必须要求的
kernel_size
(orkernel_h
andkernel_w
): 每个滤波器的高和宽
- 强烈推荐的
- weight_filter [default
type: 'constant' value: 0
]
- weight_filter [default
- 可选的
pool
[default MAX]: pooling的方法,包括MAX, AVE, or STOCHASTICpad
(orpad_h
andpad_w
) [default 0]: 指定在输入图像的每个边隐含添加的像素数目stride
(orstride_h
andstride_w
) [default 1]: 指定应用滤波器到图像时滤波器的间隔
- 必须要求的
- 输入: n * c * h_i * w_i
- 输出:n * c * h_o * w_o,其中h_o = (h_i + 2 * pad_h - kernel_h) / stride_h + 1,w_o可得类似结果。
- 例子:
layer { name: "pool1" type: "Pooling" bottom: "conv1" top: "pool1" pooling_param { pool: MAX kernel_size: 3 # pool over a 3x3 region stride: 2 # step two pixels (in the bottom blob) between pooling regions } }
Local Response Normalization (LRN): 局部响应归一化
- layer类型:LRN
- CPU实现:.
/src/caffe/layers/lrn_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/lrn_layer.cu
- 参数(
LRNParameter lrn_param
)- 可选的
local_size
[default 5]: 指跨通道LRN求和的通道数目,或者通道内LRN求和的方形区域边长alpha
[default 1]: 尺度参数beta
[default 5]: 指数norm_region
[defaultACROSS_CHANNELS
]: 在相邻通道上求和 (ACROSS_CHANNELS
) 或者在通道内附近区域中求和 (WITHIN_CHANNEL
)
- 可选的
局部响应归一化层是一种侧抑制(lateral inhibition),在局部输入区域上进行归一化。在ACROSS_CHANNELS
模式下,局部区域扩展到相邻通道,但是没有空间扩展(也就是形状是local_size * 1 * 1)。在WITHIN_CHANNEL
模式下,局部区域空间扩展,但是在各自的通道内(形状是1 * local_size * local_size)。每个输入值除以,n是每个局部区域的尺寸,求和是在以当前位置为中心的区域上操作。
im2col:它是一个做图像到列向量变换的工具,我们不需要了解。它在caffe原始的卷积中使用,通过把所有patches放入一个矩阵进行矩阵乘法。
Loss Layers:损失层
loss驱动了学习过程,它比较输出和目标之间的差异,并为之设置代价去最小化。loss本身被前向传播计算,关于loss的梯度被后向传播计算。
Softmax:type: SoftmaxWithLoss
softmax损失层计算输入的softmax的多项式logistic loss。它概念上等同于一个softmax layer,后面连接一个多项式logistic loss layer,但是softmax loss layer提供一个数值更稳定的梯度。
Sum-of-Squares / Euclidean: type: EuclideanLoss
Euclidean损失层计算两个输入的差的平方和:
Hinge / Margin: 铰链损失或边缘损失
- layer类型:HingeLoss
- CPU实现:.
/src/caffe/layers/hinge_loss_layer.cpp
- CUDA GPU实现:
目前尚无GPU实现
- 参数(
HingeLossParameter hinge_loss_param
)- 可选的
norm
[default L1]: 使用范数,目前包括 L1, L2两种选择。
- 可选的
- 输入:
n * c * h * w
Predictionsn * 1 * 1 * 1
Labels
- 输出:
1 * 1 * 1 * 1
所得损失 - 例子:
# L1 Norm layer { name: "loss" type: "HingeLoss" bottom: "pred" bottom: "label" } # L2 Norm layer { name: "loss" type: "HingeLoss" bottom: "pred" bottom: "label" top: "loss" hinge_loss_param { norm: L2 } }
Sigmoid Cross-Entropy: type: SigmoidCrossEntropyLoss交叉熵损失,用于多标签分类
Infogain: type: InfogainLoss信息增益损失
Accuracy and Top-K: 准确性对输出进行评分,计算输出与目标之间的差异,它实际上不是一个loss,没有后向传播阶段。
Activiation / Neuron Layers:激励或神经元层
通常下,这类layer都是element-wise操作,输入一个bottom blob,产生一个同样大小的blob。在下面的layer介绍中,我们忽略了输入输出大小,因为它们是相同的,都是n * c * h * w。
ReLU / Rectified-Linear and Leaky-ReLU:
- layer类型:ReLU
- CPU实现:.
/src/caffe/layers/relu_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/relu_layer.cu
- 参数(
ReLUParameter relu_param
)- 可选的
negative_slope
[default 0]: 指定是否使用斜坡值代替负数部分,还是将负数部分直接设置为0.
- 可选的
- 例子:
layer { name: "relu1" type: "ReLU" bottom: "conv1" top: "conv1" }
给定一个输入值x,ReLU层在x > 0时输出x, x < 0时输出
negative_slope
* x。当negative_slope
参数没有设置时,等价于标准ReLU函数(max(x, 0))。它支持原位运算,意味着bottom和top blob是同址的,减少了内存消耗。
Sigmoid:
- layer类型:Sigmoid
- CPU实现:.
/src/caffe/layers/sigmoid_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/sigmoid_layer.cu
- 例子:
layer { name: "encode1neuron" bottom: "encode1" top: "encode1neuron" type: "Sigmoid" }
TanH / Hyperbolic Tangent
- layer类型:TanH
- CPU实现:.
/src/caffe/layers/tanh_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/tanh_layer.cu
- 例子:
layer { name: "layer" bottom: "in" top: "out" type: "TanH" }
Absolute Value
- layer类型:AbsVal
- CPU实现:.
/src/caffe/layers/absval_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/absval_layer.cu
- 例子:
layer { name: "layer" bottom: "in" top: "out" type: "AbsVal" }
Power
- layer类型:Power
- CPU实现:.
/src/caffe/layers/power_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/power_layer.cu
- 参数(
PowerParameter power_param
)- 可选的
power
[default 1]scale
[default 1]shift
[default 0]
- 可选的
- 例子:
layer { name: "layer" bottom: "in" top: "out" type: "Power" power_param { power: 1 scale: 1 shift: 0 } }
power层计算输入为x时的,输出为(shift + scale * x)^power。
BNLL (Binomial Normal Log Likelihood) 二项式标准对数似然
- layer类型:BNLL
- CPU实现:.
/src/caffe/layers/bnll_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/bnll_layer.cu
- 例子:
layer { name: "layer" bottom: "in" top: "out" type: BNLL }
BNLL layer计算输入x的输出为log(1 + exp(x))。
Data Layers:数据层
数据进入caffe需要经过数据层,数据层位于网络的底部。数据可以来源于有效的数据库(LevelDB或LMDB),直接来源于内存,或者从磁盘文件以HDF5或通用图像格式。
通常输入预处理(减均值,尺度化,随机裁剪,镜像)可以通过TransformationParameters指定。
Database:来源于LevelDB或LMDB的数据
- layer类型:Data
参数:
- 必需的
source
: 包含数据文件的目录名batch_size
: 每次处理的输入数目
- 可选的
rand_skip
: 开始时跳过的输入数目,对异步SGDbackend
[defaultLEVELDB
]: 选择是否使用LEVELDB
或
LMDB
- 必需的
In-Memory:来源于内存的数据
- layer类型:MemoryData
参数:
- 必需的
batch_size
,channels
,height
,width
: 指定从内存中读取的输入块大小
- 必需的
内存数据层直接从内存中读取数据,不拷贝。为了使用,需要调用MemoryDataLayer::Reset (from C++) 或Net.set_input_arrays (from Python) 指定连续数据的源,例如4D行主序数组,一次读取一个batch-size的数据块。
HDF5 Input:来源于HDF5输入
- layer类型:HDF5Data
参数:
- 必需的
source
: 读取数据的文件名- batch_size
- 必需的
HDF5 Output:HDF5输出
- layer类型:HDF5Output
参数:
- 必需的
filename
: 写数据的文件名
- 必需的
Images:图像输入
- layer类型:ImageData
参数:
- 必需的
source
: 一个文本文件的名字,文件中每行给出一个图片名和label- batch_size: 每个batch处理的图像数量
- 可选的
- rand_skip
- shuffle [default false]:打乱顺序与否
- new_height, new_ 如果给出定义,将所有图像resize到这个尺寸
- 必需的
Windows type: WindowData
Dummy
DummyData
用来开发和debug, 详见 DummyDataParameter
.
Common Layers:一般层
Inner Product
- layer类型:InnerProduct
- CPU实现:.
/src/caffe/layers/inner_product_layer.cpp
- CUDA GPU实现:
./src/caffe/layers/inner_product_layer.cu
- 参数(
InnerProductParameter inner_product_param
)- 必需的
num_output (c_o): 滤波器数目
- 强烈推荐的
weight_filler
[defaulttype: 'constant' value: 0
]
- 可选的
bias_filler
[defaulttype: 'constant' value: 0
]bias_term
[defaulttrue
]: 指定是否对滤波器输出学习和应用一组附加偏差项
- 必需的
- 输入:n * c_i * h_i * w_i
- 输出:n * c_o * 1 * 1
- 例子
layer { name: "fc8" type: "InnerProduct" # learning rate and decay multipliers for the weights param { lr_mult: 1 decay_mult: 1 } # learning rate and decay multipliers for the biases param { lr_mult: 2 decay_mult: 0 } inner_product_param { num_output: 1000 weight_filler { type: "gaussian" std: 0.01 } bias_filler { type: "constant" value: 0 } } bottom: "fc7" top: "fc8" }
内积层(实际上通常指全连接层)将输入看成简单向量,产生一个单个向量形式的输出(blob的高和宽设置为1)。
Splitting:分割
分割层是一个功能层,将输入blob分成多个输出blob。这个layer用于一个blob被输入到多个输出层的情况。
Flattening:压扁
flatten layer也是一个功能层,将形为n * c * h * w的blob输入压扁成一个形为n * (c * h * w)的简单向量,实际上是单独压缩,每个数据是一个简单向量,维度c * h * w,共n个向量。
Reshape:整形
- layer类型:Reshape
- CPU实现:.
/src/caffe/layers/reshape_layer.cpp
- 参数(
ReshapeParameter reshape_param
)- 可选的
- shape
- 可选的
- 输入:一个任意维度的blob
- 输出:同一个blob,维度修改为reshape_param
- 例子:
layer { name: "reshape" type: "Reshape" bottom: "input" top: "output" reshape_param { shape { dim: 0 # copy the dimension from below dim: 2 dim: 3 dim: -1 # infer it from the other dimensions } } }
reshape layer用于改变输入维度,但是不改变数据。就像flatten layer一样,仅仅数据维度改变,过程中没有数据被拷贝。
输出维度被Reshape_param指定。帧数直接使用,设置相应的输出blob的维度。在目标维度值设置时,两个特殊值被接受:- 0: 从bottom layer拷贝相应维度。如果给定dim: 0,且bottom由2作为第一维维度,那么top layer也由2作为第一维维度 ==> 不改变原始维度
- -1:代表从其他维度推断这一维维度。这个行为与numpy的-1和Matlab reshape时的[ ]作用是相似的。维度被计算,使得总体输出维度与bottom layer相似。在reshape操作中至多可以设置一个-1。
另外一个例子,指定reshape_param{shape{dim: 0 dim:-1}}作用与Flatten layer作用相同,都是将输入blob压扁成向量。
Concatenation:拼接
concat layer是一个功能层,用于将多个输入blob拼接城一个单个的输出blob。
- layer类型:Concat
- CPU实现:.
/src/caffe/layers/concat_layer.cpp
- CUDA GPU实现:.
/src/caffe/layers/concat_layer.cu
- 参数(
ConcatParameter concat_param
)- 可选的
- axis [default 1]: 0表示沿着num连接,1表示按通道连接。
- 可选的
- 输入:n_i * c_i * h * w,K个输入blob
- 输出:
- 如果axis = 0: (n_1 + n_2 + ... + n_K) * c_1 * h * w,所有输入的c_i应该相同;
- 如果axis = 1: n_1 * (c_1 + c_2 + ... + c_K) * h * w,所有输入的n_i应该相同。
- 例子:
layer { name: "concat" bottom: "in1" bottom: "in2" top: "out" type: "Concat" concat_param { axis: 1 } }
Slicing:切片
slice layer也是一个功能层,将一个输入层沿着给定维度(当前仅提供基于num和通道的实现)切片成多个输出层。
例子:
layer { name: "slicer_label" type: "Slice" bottom: "label" ## Example of label with a shape N x 3 x 1 x 1 top: "label1" top: "label2" top: "label3" slice_param { axis: 1 slice_point: 1 slice_point: 2 } }
axis表示目标axis,沿着给定维度切片。slice_point表示选择维度的索引,索引数目应该等于顶层blob数目减一。
Elementwise Operations
Eltwise
Argmax
ArgMax
Softmax
Softmax
Mean-Variance Normalization
MVN