caffe添加自定义层级

一、整体流程

修改src/caffe/proto/caffe.proto文件->在caffe/include/caffe/layers与caffe/src/caffe/layers添加对应层级的头文件与源文件->编译caffe库

二、实现细节

第一步：在对应src/caffe/proto/caffe.proto文件找到message LayerParameter{…}的代码，并在其中添加

optional YOUR_LAYERS_NAME layers_param = 任意proto中未出现的数字;

[注意新增加行中的ID数值，可以为其他值（如200，500等），但是不能与当前已经存在的层参数ID相同。在Note中已经提示下一个可用的参数ID为147等。]
第二步：添加message AllPassParameter { }，指定参数名称

message YOUR_LAYERS_Parameter {  
  optional float key = 1 [default = 0];  
}  

第三步：将对应在caffe/include/caffe/layers与caffe/src/caffe/layers添加对应层级的头文件与源文件->编译caffe库