在caffe中添加自定义的layer

在处理某些问题时，可能会需要自己去定义特定的layer来实现某些特殊功能，现将添加新的caffe layer的具体步骤进行整理，便于以后复习。

（一）具体步骤如下：

1. 修改{caffe_dir}/src/caffe/proto/caffe.proto 文件，在 V1 LayerParameter的LayerType enum中添加新的layer的名称及ID(如：MYLAYER 38),并在V1LayerParameter中添加 MyLayer的参数说明;在message LayerParameter中添加MyLayer的参数说明，同时更新 LayerParameter的available ID(如：optional MyLayerParameter my_layer_param = 146);最后添加新Layer的参数说明。

message MyLayerParameter {
  optional uint32 para1 = 1; (1 is the ID of para1 in this layerparameter)
  optional bool para2 = 2 [default = true]; 
  optional FillerParameter para3 = 3; 
  optional float para4 = 4; 
}

2. 修改 {caffe_dir}/src/caffe/util下的upgrade_proto.cpp文件

    2.1 在const char* UpgradeV1LayerType(const V1LayerParameter_LayerType type)函数中添加如下代码，使解析prototxt时可以找到新添加的  layer。

       case V1LayerParameter_MYLAYER :  
    return "MyLayer"; 

    2.2 在bool UpgradeV1LayerParameter(const V1LayerParameter& v1_layer_param, LayerParameter* layer_param) 函数的相应位置添加对Mylayer 层参数的处理函数，否则无法解析该层相应的参数。

if (v1_layer_param.has_my_layer_param()) {
    layer_param->mutable_my_layer_param()->CopyFrom(
        v1_layer_param.my_layer_param());
   }

3. 在 {caffe_dir}/include/caffe/layers/ 文件夹中增加该layer的类的声明my_layer_layer.h文件。

4. 在 {caffe_dir}/src/caffe/layers/目录下新建.cpp和.cu文件，进行类实现。

5. 在 {caffe_dir}/src/caffe/test/中增加layer的测试代码，对所写的layer前传和反传进行测试，测试还包括速度。

（二）编译

make all -j8

Note：若报错，则根据相应的错误位置检查语法错误。

make test -j8

make runtest -j8

Note： 若报错，则根据相应的错误位置检查逻辑错误。

（补充）

如果不写test文件，为了检验该 Layer 是否能正常创建和执行  forward, backward，也可以利用新建的layer编写deploy.prototxt，然后运行 caffe time 命令并指定刚刚实现的 deploy.prototxt：

{caffe_dir}/build/tools/caffe.bin time -model deploy.prototxt