在caffe中添加自定义的layer

时间:2021-12-18 04:00:15
在处理某些问题时,可能会需要自己去定义特定的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